fix test_nodes.py

BlueBrain · Jun 26, 2023 · 5a77746 · 5a77746
1 parent 7e9bfe9
commit 5a77746
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/test_nodes/test_nodes.py b/tests/test_nodes/test_nodes.py
@@ -377,12 +377,11 @@ def test_functionality_with_separate_node_set(self):
         )
 
         with pytest.raises(BluepySnapError, match="Undefined node set"):
-            self.test_obj.get("ExtraLayer2")
+            next(self.test_obj.get("ExtraLayer2"))
 
-        pdt.assert_frame_equal(
-            self.test_obj.get(node_sets["ExtraLayer2"]),
-            self.test_obj.get("Layer2"),
-        )
+        tested = pd.concat(df for _, df in self.test_obj.get(node_sets["ExtraLayer2"]))
+        expected = pd.concat(df for _, df in self.test_obj.get("Layer2"))
+        pdt.assert_frame_equal(tested, expected)
 
     def test_pickle(self, tmp_path):
         pickle_path = tmp_path / "pickle.pkl"