chore: rename package to data-factory-testing-framework

microsoft · Dec 8, 2023 · 224d730 · 224d730
1 parent 75dc76e
commit 224d730
Show file tree

Hide file tree

Showing 94 changed files with 231 additions and 232 deletions.
diff --git a/.github/workflows/ci-cd.yml b/.github/workflows/ci-cd.yml
@@ -11,7 +11,7 @@ jobs:
       id-token: write
     environment:
       name: pypi
-      url: https://pypi.org/p/azure-data-factory-testing-framework
+      url: https://pypi.org/p/data-factory-testing-framework
     steps:
       #----------------------------------------------
       #       check-out repo and set-up python
@@ -66,7 +66,6 @@ jobs:
           poetry version 0.0.0.alpha${{ github.run_number}}
           poetry build
       - name: Publish package distributions to PyPI
-        if: github.ref == 'refs/heads/main'
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           packages-dir: ./dist
diff --git a/README.md b/README.md
@@ -42,8 +42,8 @@ Data Factory does not support unit testing out of the box. The only way to valid
 
 1. Set up an empty Python project with your favorite testing library
 2. Install the package using your preferred package manager:
-   * Pip: `pip install azure-data-factory-testing-framework`
-   * Poetry: `poetry add azure-data-factory-testing-framework`
+   * Pip: `pip install data-factory-testing-framework`
+   * Poetry: `poetry add data-factory-testing-framework`
 3. Start writing tests
 
 ## Features - Examples

diff --git a/examples/data_factory/batch_job/test_data_factory_batchjob_functional.py b/examples/data_factory/batch_job/test_data_factory_batchjob_functional.py
@@ -1,7 +1,7 @@
 import pytest
-from azure_data_factory_testing_framework.state import RunParameterType
-from azure_data_factory_testing_framework.state.run_parameter import RunParameter
-from azure_data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
+from data_factory_testing_framework.state import RunParameterType
+from data_factory_testing_framework.state.run_parameter import RunParameter
+from data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
 
 
 def test_batch_job_pipeline(request: pytest.FixtureRequest) -> None:

diff --git a/examples/data_factory/batch_job/test_data_factory_batchjob_unit.py b/examples/data_factory/batch_job/test_data_factory_batchjob_unit.py
@@ -1,13 +1,13 @@
 # flake8: noqa: E501
 import pytest
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
-from azure_data_factory_testing_framework.state import (
+from data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.state import (
     PipelineRunState,
     PipelineRunVariable,
     RunParameter,
     RunParameterType,
 )
-from azure_data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
+from data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
 
 
 @pytest.fixture

diff --git a/examples/fabric/batch_job/test_fabric_batchjob_functional.py b/examples/fabric/batch_job/test_fabric_batchjob_functional.py
@@ -1,7 +1,7 @@
 import pytest
-from azure_data_factory_testing_framework.state import RunParameterType
-from azure_data_factory_testing_framework.state.run_parameter import RunParameter
-from azure_data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
+from data_factory_testing_framework.state import RunParameterType
+from data_factory_testing_framework.state.run_parameter import RunParameter
+from data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
 
 
 def test_batch_job_pipeline(request: pytest.FixtureRequest) -> None:

diff --git a/examples/fabric/batch_job/test_fabric_batchjob_unit.py b/examples/fabric/batch_job/test_fabric_batchjob_unit.py
@@ -1,13 +1,13 @@
 # flake8: noqa: E501
 import pytest
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
-from azure_data_factory_testing_framework.state import (
+from data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.state import (
     PipelineRunState,
     PipelineRunVariable,
     RunParameter,
     RunParameterType,
 )
-from azure_data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
+from data_factory_testing_framework.test_framework import TestFramework, TestFrameworkType
 
 
 @pytest.fixture

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,5 +1,5 @@
 [tool.poetry]
-name = "azure-data-factory-testing-framework"
+name = "data-factory-testing-framework"
 version = "0.0.0"
 description = "A unit test framework that allows you to write unit and functional tests for Data Factory pipelines against the git integrated json resource files."
 authors = ["Arjen Kroezen <arjenkroezen@microsoft.com>", "Yennifer Santos <ysantos@microsoft.com>", "Jaya Kumar <kumarjaya@microsoft.com>", "Leonard Herold <lherold@microsoft.com>"]

diff --git a/src/azure_data_factory_testing_framework/functions/__init__.py b/src/azure_data_factory_testing_framework/functions/__init__.py
diff --git a/...ata_factory_testing_framework/__init__.py → ...ata_factory_testing_framework/__init__.py b/...ata_factory_testing_framework/__init__.py → ...ata_factory_testing_framework/__init__.py
diff --git a/...sting_framework/deserializers/__init__.py → ...sting_framework/deserializers/__init__.py b/...sting_framework/deserializers/__init__.py → ...sting_framework/deserializers/__init__.py
diff --git a/...ework/deserializers/_deserializer_base.py → ...ework/deserializers/_deserializer_base.py b/...ework/deserializers/_deserializer_base.py → ...ework/deserializers/_deserializer_base.py
@@ -1,10 +1,10 @@
-from azure_data_factory_testing_framework.deserializers.shared._activity_deserializer import (
+from data_factory_testing_framework.deserializers.shared._activity_deserializer import (
     _get_activity_from_activity_data,
 )
-from azure_data_factory_testing_framework.deserializers.shared._data_factory_element_replacer import (
+from data_factory_testing_framework.deserializers.shared._data_factory_element_replacer import (
     _find_and_replace_expressions_in_dict,
 )
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.models.pipeline import Pipeline
 
 
 def _parse_pipeline_from_json(name: str, json_data: dict) -> Pipeline:

diff --git a/...serializers/_deserializer_data_factory.py → ...serializers/_deserializer_data_factory.py b/...serializers/_deserializer_data_factory.py → ...serializers/_deserializer_data_factory.py
@@ -1,7 +1,7 @@
 import json
 
-from azure_data_factory_testing_framework.deserializers._deserializer_base import _parse_pipeline_from_json
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.deserializers._deserializer_base import _parse_pipeline_from_json
+from data_factory_testing_framework.models.pipeline import Pipeline
 
 
 def parse_data_factory_pipeline_from_pipeline_json(pipeline_json: str) -> Pipeline:

diff --git a/...ork/deserializers/_deserializer_fabric.py → ...ork/deserializers/_deserializer_fabric.py b/...ork/deserializers/_deserializer_fabric.py → ...ork/deserializers/_deserializer_fabric.py
@@ -1,7 +1,7 @@
 import json
 
-from azure_data_factory_testing_framework.deserializers._deserializer_base import _parse_pipeline_from_json
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.deserializers._deserializer_base import _parse_pipeline_from_json
+from data_factory_testing_framework.models.pipeline import Pipeline
 
 
 def parse_fabric_pipeline_from_pipeline_json_files(metadata_json: str, pipeline_json: str) -> Pipeline:

diff --git a/...ramework/deserializers/shared/__init__.py → ...ramework/deserializers/shared/__init__.py b/...ramework/deserializers/shared/__init__.py → ...ramework/deserializers/shared/__init__.py
diff --git a/...ializers/shared/_activity_deserializer.py → ...ializers/shared/_activity_deserializer.py b/...ializers/shared/_activity_deserializer.py → ...ializers/shared/_activity_deserializer.py
@@ -1,15 +1,15 @@
 from typing import List
 
-from azure_data_factory_testing_framework.models.activities.activity import Activity
-from azure_data_factory_testing_framework.models.activities.append_variable_activity import AppendVariableActivity
-from azure_data_factory_testing_framework.models.activities.execute_pipeline_activity import ExecutePipelineActivity
-from azure_data_factory_testing_framework.models.activities.fail_activity import FailActivity
-from azure_data_factory_testing_framework.models.activities.filter_activity import FilterActivity
-from azure_data_factory_testing_framework.models.activities.for_each_activity import ForEachActivity
-from azure_data_factory_testing_framework.models.activities.if_condition_activity import IfConditionActivity
-from azure_data_factory_testing_framework.models.activities.set_variable_activity import SetVariableActivity
-from azure_data_factory_testing_framework.models.activities.switch_activity import SwitchActivity
-from azure_data_factory_testing_framework.models.activities.until_activity import UntilActivity
+from data_factory_testing_framework.models.activities.activity import Activity
+from data_factory_testing_framework.models.activities.append_variable_activity import AppendVariableActivity
+from data_factory_testing_framework.models.activities.execute_pipeline_activity import ExecutePipelineActivity
+from data_factory_testing_framework.models.activities.fail_activity import FailActivity
+from data_factory_testing_framework.models.activities.filter_activity import FilterActivity
+from data_factory_testing_framework.models.activities.for_each_activity import ForEachActivity
+from data_factory_testing_framework.models.activities.if_condition_activity import IfConditionActivity
+from data_factory_testing_framework.models.activities.set_variable_activity import SetVariableActivity
+from data_factory_testing_framework.models.activities.switch_activity import SwitchActivity
+from data_factory_testing_framework.models.activities.until_activity import UntilActivity
 
 
 def _get_activity_from_activity_data(activity_data: dict) -> Activity:

diff --git a/.../shared/_data_factory_element_replacer.py → .../shared/_data_factory_element_replacer.py b/.../shared/_data_factory_element_replacer.py → .../shared/_data_factory_element_replacer.py
@@ -1,6 +1,6 @@
 from typing import Any
 
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
 
 
 def _find_and_replace_expressions_in_dict(obj: any, visited: list = None) -> None:

diff --git a/...rk/exceptions/activity_not_found_error.py → ...rk/exceptions/activity_not_found_error.py b/...rk/exceptions/activity_not_found_error.py → ...rk/exceptions/activity_not_found_error.py
diff --git a/.../activity_output_field_not_found_error.py → .../activity_output_field_not_found_error.py b/.../activity_output_field_not_found_error.py → .../activity_output_field_not_found_error.py
diff --git a/...ions/dataset_parameter_not_found_error.py → ...ions/dataset_parameter_not_found_error.py b/...ions/dataset_parameter_not_found_error.py → ...ions/dataset_parameter_not_found_error.py
diff --git a/...exceptions/expression_evaluation_error.py → ...exceptions/expression_evaluation_error.py b/...exceptions/expression_evaluation_error.py → ...exceptions/expression_evaluation_error.py
diff --git a/...s/expression_parameter_not_found_error.py → ...s/expression_parameter_not_found_error.py b/...s/expression_parameter_not_found_error.py → ...s/expression_parameter_not_found_error.py
diff --git a/...rk/exceptions/expression_parsing_error.py → ...rk/exceptions/expression_parsing_error.py b/...rk/exceptions/expression_parsing_error.py → ...rk/exceptions/expression_parsing_error.py
diff --git a/...ion_call_invalid_arguments_count_error.py → ...ion_call_invalid_arguments_count_error.py b/...ion_call_invalid_arguments_count_error.py → ...ion_call_invalid_arguments_count_error.py
diff --git a/...nked_service_parameter_not_found_error.py → ...nked_service_parameter_not_found_error.py b/...nked_service_parameter_not_found_error.py → ...nked_service_parameter_not_found_error.py
diff --git a/...e_activities_circular_dependency_error.py → ...e_activities_circular_dependency_error.py b/...e_activities_circular_dependency_error.py → ...e_activities_circular_dependency_error.py
diff --git a/...rk/exceptions/pipeline_not_found_error.py → ...rk/exceptions/pipeline_not_found_error.py b/...rk/exceptions/pipeline_not_found_error.py → ...rk/exceptions/pipeline_not_found_error.py
diff --git a/...ons/state_iteration_item_not_set_error.py → ...ons/state_iteration_item_not_set_error.py b/...ons/state_iteration_item_not_set_error.py → ...ons/state_iteration_item_not_set_error.py
diff --git a/.../exceptions/unsupported_function_error.py → .../exceptions/unsupported_function_error.py b/.../exceptions/unsupported_function_error.py → .../exceptions/unsupported_function_error.py
diff --git a/...e_being_evaluated_does_not_exist_error.py → ...e_being_evaluated_does_not_exist_error.py b/...e_being_evaluated_does_not_exist_error.py → ...e_being_evaluated_does_not_exist_error.py
diff --git a/...ceptions/variable_does_not_exist_error.py → ...ceptions/variable_does_not_exist_error.py b/...ceptions/variable_does_not_exist_error.py → ...ceptions/variable_does_not_exist_error.py
diff --git a/...rk/exceptions/variable_not_found_error.py → ...rk/exceptions/variable_not_found_error.py b/...rk/exceptions/variable_not_found_error.py → ...rk/exceptions/variable_not_found_error.py
diff --git a/src/data_factory_testing_framework/functions/__init__.py b/src/data_factory_testing_framework/functions/__init__.py
@@ -0,0 +1,5 @@
+from data_factory_testing_framework.functions.functions_repository import FunctionsRepository
+
+__all__ = [
+    "FunctionsRepository",
+]
diff --git a/...work/functions/_expression_transformer.py → ...work/functions/_expression_transformer.py b/...work/functions/_expression_transformer.py → ...work/functions/_expression_transformer.py
@@ -4,30 +4,30 @@
 from lark import Discard, Token, Transformer
 from lxml.etree import _Element
 
-from azure_data_factory_testing_framework.exceptions.activity_not_found_error import ActivityNotFoundError
-from azure_data_factory_testing_framework.exceptions.activity_output_field_not_found_error import (
+from data_factory_testing_framework.exceptions.activity_not_found_error import ActivityNotFoundError
+from data_factory_testing_framework.exceptions.activity_output_field_not_found_error import (
     ActivityOutputFieldNotFoundError,
 )
-from azure_data_factory_testing_framework.exceptions.dataset_parameter_not_found_error import (
+from data_factory_testing_framework.exceptions.dataset_parameter_not_found_error import (
     DatasetParameterNotFoundError,
 )
-from azure_data_factory_testing_framework.exceptions.expression_evaluation_error import (
+from data_factory_testing_framework.exceptions.expression_evaluation_error import (
     ExpressionEvaluationError,
 )
-from azure_data_factory_testing_framework.exceptions.expression_parameter_not_found_error import (
+from data_factory_testing_framework.exceptions.expression_parameter_not_found_error import (
     ExpressionParameterNotFoundError,
 )
-from azure_data_factory_testing_framework.exceptions.linked_service_parameter_not_found_error import (
+from data_factory_testing_framework.exceptions.linked_service_parameter_not_found_error import (
     LinkedServiceParameterNotFoundError,
 )
-from azure_data_factory_testing_framework.exceptions.state_iteration_item_not_set_error import (
+from data_factory_testing_framework.exceptions.state_iteration_item_not_set_error import (
     StateIterationItemNotSetError,
 )
-from azure_data_factory_testing_framework.exceptions.variable_not_found_error import VariableNotFoundError
-from azure_data_factory_testing_framework.functions.functions_repository import FunctionsRepository
-from azure_data_factory_testing_framework.state.pipeline_run_state import PipelineRunState
-from azure_data_factory_testing_framework.state.run_parameter import RunParameter
-from azure_data_factory_testing_framework.state.run_parameter_type import RunParameterType
+from data_factory_testing_framework.exceptions.variable_not_found_error import VariableNotFoundError
+from data_factory_testing_framework.functions.functions_repository import FunctionsRepository
+from data_factory_testing_framework.state.pipeline_run_state import PipelineRunState
+from data_factory_testing_framework.state.run_parameter import RunParameter
+from data_factory_testing_framework.state.run_parameter_type import RunParameterType
 
 
 class ExpressionTransformer(Transformer):

diff --git a/...amework/functions/expression_evaluator.py → ...amework/functions/expression_evaluator.py b/...amework/functions/expression_evaluator.py → ...amework/functions/expression_evaluator.py
@@ -3,10 +3,10 @@
 from lark import Lark, Token, Tree, UnexpectedCharacters
 from lark.exceptions import VisitError
 
-from azure_data_factory_testing_framework.exceptions.expression_parsing_error import ExpressionParsingError
-from azure_data_factory_testing_framework.functions._expression_transformer import ExpressionTransformer
-from azure_data_factory_testing_framework.functions.functions_repository import FunctionsRepository
-from azure_data_factory_testing_framework.state.pipeline_run_state import PipelineRunState
+from data_factory_testing_framework.exceptions.expression_parsing_error import ExpressionParsingError
+from data_factory_testing_framework.functions._expression_transformer import ExpressionTransformer
+from data_factory_testing_framework.functions.functions_repository import FunctionsRepository
+from data_factory_testing_framework.state.pipeline_run_state import PipelineRunState
 
 
 class ExpressionEvaluator:

diff --git a/...ns/functions_collection_implementation.py → ...ns/functions_collection_implementation.py b/...ns/functions_collection_implementation.py → ...ns/functions_collection_implementation.py
diff --git a/...ns/functions_conversion_implementation.py → ...ns/functions_conversion_implementation.py b/...ns/functions_conversion_implementation.py → ...ns/functions_conversion_implementation.py
diff --git a/...unctions/functions_date_implementation.py → ...unctions/functions_date_implementation.py b/...unctions/functions_date_implementation.py → ...unctions/functions_date_implementation.py
diff --git a/...tions/functions_logical_implementation.py → ...tions/functions_logical_implementation.py b/...tions/functions_logical_implementation.py → ...tions/functions_logical_implementation.py
diff --git a/...unctions/functions_math_implementation.py → ...unctions/functions_math_implementation.py b/...unctions/functions_math_implementation.py → ...unctions/functions_math_implementation.py
diff --git a/...amework/functions/functions_repository.py → ...amework/functions/functions_repository.py b/...amework/functions/functions_repository.py → ...amework/functions/functions_repository.py
@@ -1,11 +1,11 @@
 from typing import Callable, Dict
 
-import azure_data_factory_testing_framework.functions.functions_collection_implementation as collection_functions
-import azure_data_factory_testing_framework.functions.functions_conversion_implementation as conversion_functions
-import azure_data_factory_testing_framework.functions.functions_date_implementation as date_functions
-import azure_data_factory_testing_framework.functions.functions_logical_implementation as logical_functions
-import azure_data_factory_testing_framework.functions.functions_math_implementation as math_functions
-import azure_data_factory_testing_framework.functions.functions_string_implementation as string_functions
+import data_factory_testing_framework.functions.functions_collection_implementation as collection_functions
+import data_factory_testing_framework.functions.functions_conversion_implementation as conversion_functions
+import data_factory_testing_framework.functions.functions_date_implementation as date_functions
+import data_factory_testing_framework.functions.functions_logical_implementation as logical_functions
+import data_factory_testing_framework.functions.functions_math_implementation as math_functions
+import data_factory_testing_framework.functions.functions_string_implementation as string_functions
 
 
 class FunctionsRepository:

diff --git a/...ctions/functions_string_implementation.py → ...ctions/functions_string_implementation.py b/...ctions/functions_string_implementation.py → ...ctions/functions_string_implementation.py
diff --git a/...tory_testing_framework/models/__init__.py → ...tory_testing_framework/models/__init__.py b/...tory_testing_framework/models/__init__.py → ...tory_testing_framework/models/__init__.py
diff --git a/...g_framework/models/activities/__init__.py → ...g_framework/models/activities/__init__.py b/...g_framework/models/activities/__init__.py → ...g_framework/models/activities/__init__.py
diff --git a/...g_framework/models/activities/activity.py → ...g_framework/models/activities/activity.py b/...g_framework/models/activities/activity.py → ...g_framework/models/activities/activity.py
@@ -1,11 +1,11 @@
 from typing import Any, List
 
-from azure_data_factory_testing_framework.models.activities.activity_dependency import (
+from data_factory_testing_framework.models.activities.activity_dependency import (
     ActivityDependency,
 )
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
-from azure_data_factory_testing_framework.state.dependency_condition import DependencyCondition
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.state.dependency_condition import DependencyCondition
 
 
 class Activity:

diff --git a/.../models/activities/activity_dependency.py → .../models/activities/activity_dependency.py b/.../models/activities/activity_dependency.py → .../models/activities/activity_dependency.py
@@ -1,6 +1,6 @@
 from typing import List, Union
 
-from azure_data_factory_testing_framework.state.dependency_condition import DependencyCondition
+from data_factory_testing_framework.state.dependency_condition import DependencyCondition
 
 
 class ActivityDependency:

diff --git a/...ls/activities/append_variable_activity.py → ...ls/activities/append_variable_activity.py b/...ls/activities/append_variable_activity.py → ...ls/activities/append_variable_activity.py
@@ -1,8 +1,8 @@
 from typing import Any
 
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
 
 
 class AppendVariableActivity(ControlActivity):

diff --git a/...ork/models/activities/control_activity.py → ...ork/models/activities/control_activity.py b/...ork/models/activities/control_activity.py → ...ork/models/activities/control_activity.py
@@ -1,7 +1,7 @@
 from typing import Any, Callable, Iterator, List
 
-from azure_data_factory_testing_framework.models.activities.activity import Activity
-from azure_data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.models.activities.activity import Activity
+from data_factory_testing_framework.state import PipelineRunState
 
 
 class ControlActivity(Activity):

diff --git a/...s/activities/execute_pipeline_activity.py → ...s/activities/execute_pipeline_activity.py b/...s/activities/execute_pipeline_activity.py → ...s/activities/execute_pipeline_activity.py
@@ -1,11 +1,11 @@
 from typing import Any, Callable, Iterator, List
 
-from azure_data_factory_testing_framework.models.activities.activity import Activity
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.models.pipeline import Pipeline
-from azure_data_factory_testing_framework.state import PipelineRunState, RunParameterType
-from azure_data_factory_testing_framework.state.run_parameter import RunParameter
+from data_factory_testing_framework.models.activities.activity import Activity
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.models.pipeline import Pipeline
+from data_factory_testing_framework.state import PipelineRunState, RunParameterType
+from data_factory_testing_framework.state.run_parameter import RunParameter
 
 
 class ExecutePipelineActivity(ControlActivity):

diff --git a/...mework/models/activities/fail_activity.py → ...mework/models/activities/fail_activity.py b/...mework/models/activities/fail_activity.py → ...mework/models/activities/fail_activity.py
@@ -1,8 +1,8 @@
 from typing import Any
 
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.state import PipelineRunState
-from azure_data_factory_testing_framework.state.dependency_condition import DependencyCondition
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.state.dependency_condition import DependencyCondition
 
 
 class FailActivity(ControlActivity):

diff --git a/...work/models/activities/filter_activity.py → ...work/models/activities/filter_activity.py b/...work/models/activities/filter_activity.py → ...work/models/activities/filter_activity.py
@@ -1,9 +1,9 @@
 from typing import Any
 
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
-from azure_data_factory_testing_framework.state.dependency_condition import DependencyCondition
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.state.dependency_condition import DependencyCondition
 
 
 class FilterActivity(ControlActivity):

diff --git a/...rk/models/activities/for_each_activity.py → ...rk/models/activities/for_each_activity.py b/...rk/models/activities/for_each_activity.py → ...rk/models/activities/for_each_activity.py
@@ -1,9 +1,9 @@
 from typing import Any, Callable, Iterator, List
 
-from azure_data_factory_testing_framework.models.activities.activity import Activity
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.models.activities.activity import Activity
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
 
 
 class ForEachActivity(ControlActivity):

diff --git a/...odels/activities/if_condition_activity.py → ...odels/activities/if_condition_activity.py b/...odels/activities/if_condition_activity.py → ...odels/activities/if_condition_activity.py
@@ -1,9 +1,9 @@
 from typing import Any, Callable, Iterator, List
 
-from azure_data_factory_testing_framework.models.activities.activity import Activity
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.models.activities.activity import Activity
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
 
 
 class IfConditionActivity(ControlActivity):

diff --git a/...odels/activities/set_variable_activity.py → ...odels/activities/set_variable_activity.py b/...odels/activities/set_variable_activity.py → ...odels/activities/set_variable_activity.py
@@ -1,8 +1,8 @@
 from typing import Any
 
-from azure_data_factory_testing_framework.models.activities.control_activity import ControlActivity
-from azure_data_factory_testing_framework.models.data_factory_element import DataFactoryElement
-from azure_data_factory_testing_framework.state import PipelineRunState
+from data_factory_testing_framework.models.activities.control_activity import ControlActivity
+from data_factory_testing_framework.models.data_factory_element import DataFactoryElement
+from data_factory_testing_framework.state import PipelineRunState
 
 
 class SetVariableActivity(ControlActivity):