add test for top/bottom names

1 files changed, 13 insertions, 0 deletions

diff --git a/python/caffe/test/test_net.py b/python/caffe/test/test_net.py
index 4cacfcd0..96821e40 100644
--- a/python/caffe/test/test_net.py
+++ b/python/caffe/test/test_net.py
@@ -3,6 +3,7 @@ import tempfile
 import os
 import numpy as np
 import six
+from collections import OrderedDict
 
 import caffe
 
@@ -67,6 +68,18 @@ class TestNet(unittest.TestCase):
         self.assertEqual(self.net.inputs, [])
         self.assertEqual(self.net.outputs, ['loss'])
 
+    def test_top_bottom_names(self):
+        self.assertEqual(self.net.top_names,
+                         OrderedDict([('data', ['data', 'label']),
+                                      ('conv', ['conv']),
+                                      ('ip', ['ip']),
+                                      ('loss', ['loss'])]))
+        self.assertEqual(self.net.bottom_names,
+                         OrderedDict([('data', []),
+                                      ('conv', ['data']),
+                                      ('ip', ['conv']),
+                                      ('loss', ['ip', 'label'])]))
+
     def test_save_and_read(self):
         f = tempfile.NamedTemporaryFile(mode='w+', delete=False)
         f.close()