test for overzoom

Author: martham93

test/fixtures/integration/config_3way.integration.json

@@ -19,5 +19,4 @@
   "ml_type": "classification",
   "seed": 19,
   "split_names": ["train", "test", "val"],
-  "split_vals": [0.7, 0.2, 0.1]
-}
+  "split_vals": [0.7, 0.2, 0.1]}

test/fixtures/integration/config_overzoom.integration.json

@@ -0,0 +1,24 @@
+{"country": "portugal",
+  "bounding_box": [
+    -9.4575,
+    38.8467,
+    -9.4510,
+    38.8513
+  ],
+  "zoom": 17,
+  "classes": [
+    { "name": "Water Tower", "filter": ["==", "man_made", "water_tower"] },
+    { "name": "Building", "filter": ["has", "building"] },
+    { "name": "Farmland", "filter": ["==", "landuse", "farmland"] },
+    { "name": "Ruins", "filter": ["==", "historic", "ruins"] },
+    { "name": "Parking", "filter": ["==", "amenity", "parking"] },
+    { "name": "Roads", "filter": ["has", "highway"] }
+  ],
+  "imagery": "https://api.mapbox.com/v4/mapbox.satellite/{z}/{x}/{y}.jpg?access_token=TOKEN",
+  "background_ratio": 1,
+  "ml_type": "classification",
+  "seed": 19,
+  "split_names": ["train", "test", "val"],
+  "split_vals": [0.7, 0.2, 0.1],
+  "over_zoom": 1
+}

test/fixtures/integration/tiles_overzoom/62092-50162-17.jpg

@@ -21,10 +21,15 @@ def setUpClass(cls):
         copyfile('test/fixtures/integration/labels-cl.npz', 'integration-cl-split/labels.npz')
         copytree('test/fixtures/integration/tiles', 'integration-cl-split/tiles')
 
+        makedirs('integration-cl-overzoom')
+        copyfile('test/fixtures/integration/labels-cl.npz', 'integration-cl-overzoom/labels.npz')
+        copytree('test/fixtures/integration/tiles_overzoom', 'integration-cl-overzoom/tiles')
+
     @classmethod
     def tearDownClass(cls):
         rmtree('integration-cl')
         rmtree('integration-cl-split')
+        rmtree('integration-cl-overzoom')
 
     def test_cli(self):
         """Verify data.npz produced by CLI"""
@@ -73,4 +78,16 @@ def test_cli_3way_split(self):
         # validate label data with shapes
         self.assertEqual(data['y_train'].shape, (5, 7))
         self.assertEqual(data['y_test'].shape, (2, 7))
-        self.assertEqual(data['y_val'].shape, (1, 7))
+        self.assertEqual(data['y_val'].shape, (1, 7))
+
+    def test_overzoom(self):
+        """Verify data.npz produced by CLI when overzoom is used"""
+        cmd = 'label-maker package --dest integration-cl-overzoom --config test/fixtures/integration/config_overzoom.integration.json'
+        cmd = cmd.split(' ')
+        subprocess.run(cmd, universal_newlines=True)
+
+        data = np.load('integration-cl-overzoom/data.npz')
+
+        self.assertEqual(data['x_train'].shape, (6, 512, 512, 3))
+        self.assertEqual(data['x_test'].shape, (2, 512, 512, 3))
+        self.assertEqual(data['x_val'].shape, (1, 512, 512, 3))