Commit ff64c849 authored by Thibaut Mattio's avatar Thibaut Mattio
Browse files

Make tests compatible with Python 3

* Change dictionnaries iteritems() to items()
parent 252fffb2
...@@ -271,7 +271,7 @@ class FasterRCNNMetaArchTestBase(tf.test.TestCase): ...@@ -271,7 +271,7 @@ class FasterRCNNMetaArchTestBase(tf.test.TestCase):
self.assertEqual(set(prediction_out.keys()), expected_output_keys) self.assertEqual(set(prediction_out.keys()), expected_output_keys)
self.assertAllEqual(prediction_out['image_shape'], input_image_shape) self.assertAllEqual(prediction_out['image_shape'], input_image_shape)
for output_key, expected_shape in expected_output_shapes.iteritems(): for output_key, expected_shape in expected_output_shapes.items():
self.assertAllEqual(prediction_out[output_key].shape, expected_shape) self.assertAllEqual(prediction_out[output_key].shape, expected_shape)
# Check that anchors are clipped to window. # Check that anchors are clipped to window.
......
...@@ -63,7 +63,7 @@ class MultiResolutionFeatureMapGeneratorTest(tf.test.TestCase): ...@@ -63,7 +63,7 @@ class MultiResolutionFeatureMapGeneratorTest(tf.test.TestCase):
sess.run(init_op) sess.run(init_op)
out_feature_maps = sess.run(feature_maps) out_feature_maps = sess.run(feature_maps)
out_feature_map_shapes = dict( out_feature_map_shapes = dict(
(key, value.shape) for key, value in out_feature_maps.iteritems()) (key, value.shape) for key, value in out_feature_maps.items())
self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes) self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes)
def test_get_expected_feature_map_shapes_with_inception_v3(self): def test_get_expected_feature_map_shapes_with_inception_v3(self):
...@@ -93,7 +93,7 @@ class MultiResolutionFeatureMapGeneratorTest(tf.test.TestCase): ...@@ -93,7 +93,7 @@ class MultiResolutionFeatureMapGeneratorTest(tf.test.TestCase):
sess.run(init_op) sess.run(init_op)
out_feature_maps = sess.run(feature_maps) out_feature_maps = sess.run(feature_maps)
out_feature_map_shapes = dict( out_feature_map_shapes = dict(
(key, value.shape) for key, value in out_feature_maps.iteritems()) (key, value.shape) for key, value in out_feature_maps.items())
self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes) self.assertDictEqual(out_feature_map_shapes, expected_feature_map_shapes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment