update tests YAMLS_DIR variable

nitaibezerra · Apr 19, 2023 · d7f0047 · d7f0047
1 parent 0fc2bc5
commit d7f0047
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -44,7 +44,7 @@ def dag_gen() -> DouDigestDagGenerator:
 
 @pytest.fixture()
 def yaml_parser()-> YAMLParser:
-    filepath = os.path.join(DouDigestDagGenerator().YAMLS_DIR,
+    filepath = os.path.join(DouDigestDagGenerator().YAMLS_START_DIR,
                             'basic_example.yaml')
     return YAMLParser(filepath=filepath)
 

diff --git a/tests/parsers_test.py b/tests/parsers_test.py
@@ -180,7 +180,7 @@ def test_hash_dag_id(yaml_parser, dag_id, size, hashed):
         ),
     ])
 def test_parse(filepath, result_tuple):
-    filepath = os.path.join(DouDigestDagGenerator().YAMLS_DIR, filepath)
+    filepath = os.path.join(DouDigestDagGenerator().YAMLS_START_DIR, filepath)
     parsed = YAMLParser(filepath=filepath).parse()
 
     assert parsed == DAGConfig(**result_tuple)