fix name test broken by switch to module for tests

eacharles · Jan 20, 2021 · 0c64414 · 0c64414
1 parent 646731b
commit 0c64414
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_stage.py b/tests/test_stage.py
@@ -31,7 +31,7 @@ class TestStage(PipelineStage):
         config = {}
 
     assert PipelineStage.get_stage("test") == TestStage
-    assert TestStage.get_module() == "test_stage"
+    assert TestStage.get_module().endswith("test_stage")
 
     s = TestStage({"config": "tests/config.yml"})