Moving the Base Test Helper (#1485)

Summary:
Pull Request resolved: #1485

moving the base test helper to the captum library. This has a lot of file dependency so apologies for the large Diff

Reviewed By: cyrjano, lurunming

Differential Revision: D68174196

fbshipit-source-id: 84f180748bd6c70cba1b7079a14e97e520228feb

Author: styusuf

tests/helpers/__init__.py renamed to captum/testing/helpers/__init__.py

@@ -3,7 +3,7 @@
 # pyre-strict
 
 try:
-    from tests.helpers.fb.internal_base import FbBaseTest as BaseTest
+    from captum.testing.helpers.fb.internal_base import FbBaseTest as BaseTest
 
     __all__ = [
         "BaseTest",
@@ -13,4 +13,4 @@
     # tests/helpers/__init__.py:13: error: Incompatible import of "BaseTest"
     # (imported name has type "type[BaseTest]", local name has type
     # "type[FbBaseTest]")  [assignment]
-    from tests.helpers.basic import BaseTest  # type: ignore
+    from captum.testing.helpers.basic import BaseTest  # type: ignore

tests/helpers/basic.py renamed to captum/testing/helpers/basic.py

@@ -4,7 +4,7 @@
 from typing import Tuple, Union
 
 import torch
-from tests.helpers import BaseTest
+from captum.testing.helpers import BaseTest
 from torch import Tensor
 
 

tests/helpers/basic_models.py renamed to captum/testing/helpers/basic_models.py

@@ -5,7 +5,7 @@
 
 import torch
 from captum._utils.gradient import compute_gradients
-from tests.helpers.basic_models import BasicModel, BasicModel5_MultiArgs
+from captum.testing.helpers.basic_models import BasicModel, BasicModel5_MultiArgs
 from torch import Tensor
 from torch.nn import Module
 

tests/helpers/classification_models.py renamed to captum/testing/helpers/classification_models.py

@@ -40,8 +40,8 @@
 from captum.attr._core.saliency import Saliency
 from captum.attr._core.shapley_value import ShapleyValueSampling
 from captum.attr._utils.input_layer_wrapper import ModelInputWrapper
-from tests.helpers.basic import set_all_random_seeds
-from tests.helpers.basic_models import (
+from captum.testing.helpers.basic import set_all_random_seeds
+from captum.testing.helpers.basic_models import (
     BasicModel_ConvNet,
     BasicModel_MultiLayer,
     BasicModel_MultiLayer_MultiInput,

tests/helpers/evaluate_linear_model.py renamed to captum/testing/helpers/evaluate_linear_model.py

@@ -8,13 +8,13 @@
 import torch
 from captum._utils.typing import TensorLikeList
 from captum.attr._core.layer.grad_cam import LayerGradCam
-from packaging import version
-from tests.helpers import BaseTest
-from tests.helpers.basic import assertTensorTuplesAlmostEqual
-from tests.helpers.basic_models import (
+from captum.testing.helpers import BaseTest
+from captum.testing.helpers.basic import assertTensorTuplesAlmostEqual
+from captum.testing.helpers.basic_models import (
     BasicModel_ConvNet_One_Conv,
     BasicModel_MultiLayer,
 )
+from packaging import version
 from torch import Tensor
 from torch.nn import Module
 

tests/attr/helpers/attribution_delta_util.py

@@ -7,12 +7,12 @@
 import torch
 from captum._utils.typing import BaselineType
 from captum.attr._core.layer.internal_influence import InternalInfluence
-from packaging import version
-from tests.helpers.basic import assertTensorTuplesAlmostEqual, BaseTest
-from tests.helpers.basic_models import (
+from captum.testing.helpers.basic import assertTensorTuplesAlmostEqual, BaseTest
+from captum.testing.helpers.basic_models import (
     BasicModel_MultiLayer,
     BasicModel_MultiLayer_MultiInput,
 )
+from packaging import version
 from torch import Tensor
 from torch.nn import Module