Add test for scale_shape.

toshihiroryuu · Aug 24, 2017 · c99abf3 · c99abf3
1 parent 9a97b76
commit c99abf3
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/preprocessing/test_object_detection.py b/tests/preprocessing/test_object_detection.py
@@ -0,0 +1,16 @@
+import keras_rcnn.preprocessing._object_detection
+import numpy
+
+
+def test_scale_shape():
+    min_size = 200
+    max_size = 300
+    shape    = (600, 1000, 3)
+
+    shape, scale = keras_rcnn.preprocessing._object_detection.scale_shape(shape, min_size, max_size)
+
+    expected = (180, 300, 3)
+    numpy.testing.assert_equal(shape, expected)
+
+    expected = 0.3
+    assert numpy.isclose(scale, expected)