add tests for pycaffe's layer_dict

BVLC · Jun 22, 2016 · 5417f10 · 5417f10
1 parent b29d271
commit 5417f10
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/python/caffe/test/test_net.py b/python/caffe/test/test_net.py
@@ -59,6 +59,13 @@ def test_memory(self):
         for bl in blobs:
             total += bl.data.sum() + bl.diff.sum()
 
+    def test_layer_dict(self):
+        layer_dict = self.net.layer_dict
+        self.assertEqual(list(layer_dict.keys()), list(self.net._layer_names))
+        for i, name in enumerate(self.net._layer_names):
+            self.assertEqual(layer_dict[name].type,
+                             self.net.layers[i].type)
+
     def test_forward_backward(self):
         self.net.forward()
         self.net.backward()