diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_graph.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tests/migrations/test_graph.py b/tests/migrations/test_graph.py index 0e26f70cb6..3fcdd12667 100644 --- a/tests/migrations/test_graph.py +++ b/tests/migrations/test_graph.py @@ -174,7 +174,8 @@ class GraphTests(TestCase): graph.add_node(("app_b", "0001"), None) graph.add_dependency("app_a.0003", ("app_a", "0003"), ("app_a", "0002")) graph.add_dependency("app_a.0002", ("app_a", "0002"), ("app_a", "0001")) - with self.assertRaisesMessage(KeyError, "Migration app_a.0001 dependencies references nonexistent parent node ('app_b', '0002')"): + msg = "Migration app_a.0001 dependencies reference nonexistent parent node ('app_b', '0002')" + with self.assertRaisesMessage(KeyError, msg): graph.add_dependency("app_a.0001", ("app_a", "0001"), ("app_b", "0002")) def test_missing_child_nodes(self): @@ -184,5 +185,6 @@ class GraphTests(TestCase): # Build graph graph = MigrationGraph() graph.add_node(("app_a", "0001"), None) - with self.assertRaisesMessage(KeyError, "Migration app_a.0002 dependencies references nonexistent child node ('app_a', '0002')"): + msg = "Migration app_a.0002 dependencies reference nonexistent child node ('app_a', '0002')" + with self.assertRaisesMessage(KeyError, msg): graph.add_dependency("app_a.0002", ("app_a", "0002"), ("app_a", "0001")) |