diff options
| -rw-r--r-- | tests/gis_tests/rasterapp/test_rasterfield.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/gis_tests/rasterapp/test_rasterfield.py b/tests/gis_tests/rasterapp/test_rasterfield.py index 794c67ec61..b9467f1452 100644 --- a/tests/gis_tests/rasterapp/test_rasterfield.py +++ b/tests/gis_tests/rasterapp/test_rasterfield.py @@ -68,7 +68,10 @@ class RasterFieldTest(TransactionTestCase): rast=Func(F('rast'), function='ST_SetBandIsNoData'), ) r.refresh_from_db() - self.assertEqual(r.rast.bands[0].data(), [[no_data]]) + band = r.rast.bands[0].data() + if numpy: + band = band.flatten().tolist() + self.assertEqual(band, [no_data]) self.assertEqual(r.rast.bands[0].nodata_value, no_data) def test_model_creation(self): |