update tests since now we can deterministically predict the output ma…

…pping
Qiskit · Jul 15, 2021 · e39dcbe · e39dcbe
1 parent 8f3c701
commit e39dcbe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 42 deletions.
diff --git a/tests/digraph/test_isomorphic.py b/tests/digraph/test_isomorphic.py
@@ -259,11 +259,7 @@ def test_digraph_vf2_mapping_identical_vf2pp(self):
         mapping = retworkx.digraph_vf2_mapping(
             graph, second_graph, id_order=False
         )
-        valid_mappings = [
-            {0: 0, 1: 1, 2: 2, 3: 3},
-            {0: 0, 1: 2, 2: 1, 3: 3},
-        ]
-        self.assertIn(next(mapping), valid_mappings)
+        self.assertEqual(next(mapping), {0: 0, 1: 1, 2: 2, 3: 3})
 
     def test_graph_vf2_mapping_identical_removals_vf2pp(self):
         graph = retworkx.generators.directed_path_graph(2)
@@ -291,11 +287,7 @@ def test_subgraph_vf2_mapping_vf2pp(self):
         mapping = retworkx.digraph_vf2_mapping(
             graph, second_graph, subgraph=True, id_order=False
         )
-        valid_mappings = [
-            {8: 3, 5: 2, 7: 1, 4: 0},
-            {7: 2, 5: 1, 4: 0, 8: 3},
-        ]
-        self.assertIn(next(mapping), valid_mappings)
+        self.assertEqual(next(mapping), {4: 0, 5: 1, 7: 2, 8: 3})
 
     def test_vf2pp_remapping(self):
         temp = retworkx.generators.directed_grid_graph(3, 3)
@@ -310,8 +302,4 @@ def test_vf2pp_remapping(self):
         mapping = retworkx.digraph_vf2_mapping(
             graph, second_graph, subgraph=True, id_order=False
         )
-        expected_mappings = [
-            {6: 1, 5: 0, 8: 2, 9: 3},
-            {6: 2, 5: 0, 9: 3, 8: 1},
-        ]
-        self.assertIn(next(mapping), expected_mappings)
+        self.assertEqual(next(mapping), {5: 0, 6: 1, 8: 2, 9: 3})
diff --git a/tests/graph/test_isomorphic.py b/tests/graph/test_isomorphic.py
@@ -266,11 +266,7 @@ def test_graph_vf2_mapping_identical_vf2pp(self):
         mapping = retworkx.graph_vf2_mapping(
             graph, second_graph, id_order=False
         )
-        valid_mappings = [
-            {0: 0, 1: 1, 2: 2, 3: 3},
-            {0: 0, 1: 2, 2: 1, 3: 3},
-        ]
-        self.assertIn(next(mapping), valid_mappings)
+        self.assertEqual(next(mapping), {0: 0, 1: 1, 2: 2, 3: 3})
 
     def test_graph_vf2_mapping_identical_removals_vf2pp(self):
         graph = retworkx.generators.path_graph(2)
@@ -298,17 +294,7 @@ def test_subgraph_vf2_mapping_vf2pp(self):
         mapping = retworkx.graph_vf2_mapping(
             graph, second_graph, subgraph=True, id_order=False
         )
-        valid_mappings = [
-            {3: 2, 4: 3, 6: 0, 7: 1},
-            {3: 1, 4: 3, 6: 0, 7: 2},
-            {4: 3, 5: 1, 7: 2, 8: 0},
-            {0: 0, 1: 1, 3: 2, 4: 3},
-            {7: 1, 8: 0, 4: 3, 5: 2},
-            {5: 1, 2: 0, 1: 2, 4: 3},
-            {3: 1, 0: 0, 4: 3, 1: 2},
-            {1: 1, 2: 0, 4: 3, 5: 2},
-        ]
-        self.assertIn(next(mapping), valid_mappings)
+        self.assertEqual(next(mapping), {4: 0, 3: 2, 0: 3, 1: 1})
 
     def test_vf2pp_remapping(self):
         temp = retworkx.generators.grid_graph(3, 3)
@@ -323,14 +309,4 @@ def test_vf2pp_remapping(self):
         mapping = retworkx.graph_vf2_mapping(
             graph, second_graph, subgraph=True, id_order=False
         )
-        expected_mappings = [
-            {2: 2, 3: 0, 5: 3, 6: 1},
-            {5: 3, 6: 1, 8: 2, 9: 0},
-            {2: 1, 5: 3, 6: 2, 3: 0},
-            {2: 2, 1: 0, 5: 3, 4: 1},
-            {5: 3, 6: 2, 8: 1, 9: 0},
-            {4: 2, 1: 0, 5: 3, 2: 1},
-            {5: 3, 8: 1, 4: 2, 7: 0},
-            {5: 3, 4: 1, 7: 0, 8: 2},
-        ]
-        self.assertIn(next(mapping), expected_mappings)
+        self.assertEqual(next(mapping), {5: 0, 4: 2, 1: 3, 2: 1})