ndx_extensions tests

NovelaNeuro · Feb 13, 2020 · 7b9f00a · 7b9f00a
1 parent c8f17c4
commit 7b9f00a
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/pynwb/tests/test_apparatus.py b/src/pynwb/tests/test_apparatus.py
@@ -7,6 +7,7 @@
 
 
 class TestApparatus(TestCase):
+
     @classmethod
     def setUpClass(cls):
         cls.node_1 = Mock()
@@ -35,11 +36,14 @@ def test_successfulEdgeCreation_true(self):
 
     def test_checkEdgeCorrectValue_true(self):
         self.assertEqual(self.apparatus.name, 'ApparatusName')
-        self.assertEqual(self.apparatus.edges, {self.edge_1.name: self.edge_1, self.edge_2.name: self.edge_2})
-        self.assertEqual(self.apparatus.nodes, {self.node_1.name: self.node_1, self.node_2.name: self.node_2,
-                                                self.node_3.name: self.node_3, self.node_4.name: self.node_4})
+        self.assertEqual(self.apparatus.edges, {self.edge_1.name: self.edge_1,
+                                                self.edge_2.name: self.edge_2})
+        self.assertEqual(self.apparatus.nodes, {self.node_1.name: self.node_1,
+                                                self.node_2.name: self.node_2,
+                                                self.node_3.name: self.node_3,
+                                                self.node_4.name: self.node_4})
 
     def test_checkEdgeCorrectType_true(self):
         self.assertIsInstance(self.apparatus.name, str)
         self.assertIsInstance(self.apparatus.edges, dict)
-        self.assertIsInstance(self.apparatus.nodes, dict)
+        self.assertIsInstance(self.apparatus.nodes, dict)