@@ -654,7 +654,7 @@ def test_convert_5_multiband_to_raster(self):
654
654
)
655
655
656
656
def test_standardize_raster_for_writing_jpeg_3band (self ):
657
- raster = self .random .integers (255 , size = (3 , 5 , 6 ))
657
+ raster = self .random .integers (255 , size = (3 , 5 , 6 ), dtype = 'uint8' )
658
658
count = 'irrelevant'
659
659
driver = 'JPEG'
660
660
expected_raster = np .copy (raster )
@@ -667,7 +667,7 @@ def test_standardize_raster_for_writing_jpeg_3band(self):
667
667
np .testing .assert_array_equal (expected_raster , actual_raster , strict = True )
668
668
669
669
def test_standardize_raster_for_writing_jpeg_4band (self ):
670
- raster = self .random .integers (255 , size = (4 , 7 , 8 ))
670
+ raster = self .random .integers (255 , size = (4 , 7 , 8 ), dtype = 'uint8' )
671
671
driver = 'JPEG'
672
672
count = 'irrelevant'
673
673
expected_raster = np .copy (raster [0 :3 , :, :])
@@ -680,7 +680,7 @@ def test_standardize_raster_for_writing_jpeg_4band(self):
680
680
681
681
@patch ('hybig.browse.palettize_raster' )
682
682
def test_standardize_raster_for_writing_png_4band (self , palettize_mock ):
683
- raster = self .random .integers (255 , size = (4 , 7 , 8 ))
683
+ raster = self .random .integers (255 , size = (4 , 7 , 8 ), dtype = 'uint8' )
684
684
driver = 'PNG'
685
685
count = 'not 1'
686
686
@@ -690,7 +690,7 @@ def test_standardize_raster_for_writing_png_4band(self, palettize_mock):
690
690
691
691
@patch ('hybig.browse.palettize_raster' )
692
692
def test_standardize_raster_for_writing_png_3band (self , palettize_mock ):
693
- raster = self .random .integers (255 , size = (3 , 7 , 8 ))
693
+ raster = self .random .integers (255 , size = (3 , 7 , 8 ), dtype = 'uint8' )
694
694
driver = 'PNG'
695
695
count = 'not 1'
696
696
@@ -700,7 +700,7 @@ def test_standardize_raster_for_writing_png_3band(self, palettize_mock):
700
700
701
701
@patch ('hybig.browse.palettize_raster' )
702
702
def test_prepare_1band_raster_for_writing_png (self , palettize_mock ):
703
- raster = self .random .integers (255 , size = (1 , 7 , 8 ))
703
+ raster = self .random .integers (255 , size = (1 , 7 , 8 ), dtype = 'uint8' )
704
704
driver = 'PNG'
705
705
count = 1
706
706
palettize_mock .return_value = (None , None )
@@ -711,7 +711,7 @@ def test_prepare_1band_raster_for_writing_png(self, palettize_mock):
711
711
@patch ('hybig.browse.get_color_map_from_image' )
712
712
def test_palettize_raster_no_alpha_layer (self , get_color_map_mock , image_mock ):
713
713
"""Test that the quantize function is called by a correct image."""
714
- raster = self .random .integers (255 , dtype = 'uint8' , size = (3 , 10 , 11 ))
714
+ raster = self .random .integers (255 , size = (3 , 10 , 11 ), dtype = 'uint8' )
715
715
716
716
quantized_output = Image .fromarray (
717
717
self .random .integers (254 , size = (10 , 11 ), dtype = 'uint8' )
@@ -733,7 +733,7 @@ def test_palettize_raster_no_alpha_layer(self, get_color_map_mock, image_mock):
733
733
@patch ('hybig.browse.get_color_map_from_image' )
734
734
def test_palettize_raster_with_alpha_layer (self , get_color_map_mock , image_mock ):
735
735
"""Test that the quantize function is called by a correct image."""
736
- raster = self .random .integers (255 , dtype = 'uint8' , size = (4 , 10 , 11 ))
736
+ raster = self .random .integers (255 , size = (4 , 10 , 11 ), dtype = 'uint8' )
737
737
# No transparent pixels
738
738
raster [3 , :, :] = 255
739
739
0 commit comments