|
10 | 10 | from django.contrib.gis.gdal.error import GDALException |
11 | 11 | from django.contrib.gis.gdal.raster.band import GDALBand |
12 | 12 | from django.contrib.gis.shortcuts import numpy |
| 13 | +from django.core.files.temp import NamedTemporaryFile |
13 | 14 | from django.test import SimpleTestCase |
14 | 15 |
|
15 | 16 | from ..data.rasters.textrasters import JSON_RASTER |
@@ -148,7 +149,7 @@ def test_memory_based_raster_creation(self): |
148 | 149 |
|
149 | 150 | def test_file_based_raster_creation(self): |
150 | 151 | # Prepare tempfile |
151 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 152 | + rstfile = NamedTemporaryFile(suffix=".tif") |
152 | 153 |
|
153 | 154 | # Create file-based raster from scratch |
154 | 155 | GDALRaster( |
@@ -264,7 +265,7 @@ def test_vsi_buffer_property(self): |
264 | 265 | self.assertIsNone(self.rs.vsi_buffer) |
265 | 266 |
|
266 | 267 | def test_vsi_vsizip_filesystem(self): |
267 | | - rst_zipfile = tempfile.NamedTemporaryFile(suffix=".zip") |
| 268 | + rst_zipfile = NamedTemporaryFile(suffix=".zip") |
268 | 269 | with zipfile.ZipFile(rst_zipfile, mode="w") as zf: |
269 | 270 | zf.write(self.rs_path, "raster.tif") |
270 | 271 | rst_path = "/vsizip/" + os.path.join(rst_zipfile.name, "raster.tif") |
@@ -406,7 +407,7 @@ def test_raster_info_accessor(self): |
406 | 407 | self.assertIn("NAD83 / Florida GDL Albers", infos) |
407 | 408 |
|
408 | 409 | def test_compressed_file_based_raster_creation(self): |
409 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 410 | + rstfile = NamedTemporaryFile(suffix=".tif") |
410 | 411 | # Make a compressed copy of an existing raster. |
411 | 412 | compressed = self.rs.warp( |
412 | 413 | {"papsz_options": {"compress": "packbits"}, "name": rstfile.name} |
@@ -554,7 +555,7 @@ def test_raster_warp_nodata_zone(self): |
554 | 555 | self.assertEqual(result, [23] * 16) |
555 | 556 |
|
556 | 557 | def test_raster_clone(self): |
557 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 558 | + rstfile = NamedTemporaryFile(suffix=".tif") |
558 | 559 | tests = [ |
559 | 560 | ("MEM", "", 23), # In memory raster. |
560 | 561 | ("tif", rstfile.name, 99), # In file based raster. |
@@ -600,7 +601,7 @@ def test_raster_transform(self): |
600 | 601 | for srs in tests: |
601 | 602 | with self.subTest(srs=srs): |
602 | 603 | # Prepare tempfile and nodata value. |
603 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 604 | + rstfile = NamedTemporaryFile(suffix=".tif") |
604 | 605 | ndv = 99 |
605 | 606 | # Create in file based raster. |
606 | 607 | source = GDALRaster( |
@@ -701,7 +702,7 @@ def test_raster_transform(self): |
701 | 702 | def test_raster_transform_clone(self): |
702 | 703 | with mock.patch.object(GDALRaster, "clone") as mocked_clone: |
703 | 704 | # Create in file based raster. |
704 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 705 | + rstfile = NamedTemporaryFile(suffix=".tif") |
705 | 706 | source = GDALRaster( |
706 | 707 | { |
707 | 708 | "datatype": 1, |
@@ -729,7 +730,7 @@ def test_raster_transform_clone(self): |
729 | 730 |
|
730 | 731 | def test_raster_transform_clone_name(self): |
731 | 732 | # Create in file based raster. |
732 | | - rstfile = tempfile.NamedTemporaryFile(suffix=".tif") |
| 733 | + rstfile = NamedTemporaryFile(suffix=".tif") |
733 | 734 | source = GDALRaster( |
734 | 735 | { |
735 | 736 | "datatype": 1, |
|
0 commit comments