diff --git a/tests/unit/potential/builtin/test_bar.py b/tests/unit/potential/builtin/test_bar.py index e63af621..39274b6c 100644 --- a/tests/unit/potential/builtin/test_bar.py +++ b/tests/unit/potential/builtin/test_bar.py @@ -5,7 +5,7 @@ import galax.potential as gp -from ..test_core import TestAbstractPotential +from ..test_core import TestAbstractPotential as AbstractPotential_Test from .test_common import ( MassParameterMixin, ShapeAParameterMixin, @@ -15,7 +15,7 @@ class TestBarPotential( - TestAbstractPotential, + AbstractPotential_Test, # Parameters MassParameterMixin, ShapeAParameterMixin, diff --git a/tests/unit/potential/builtin/test_hernquist.py b/tests/unit/potential/builtin/test_hernquist.py index 016ff19e..d0306311 100644 --- a/tests/unit/potential/builtin/test_hernquist.py +++ b/tests/unit/potential/builtin/test_hernquist.py @@ -5,12 +5,12 @@ from galax.potential import HernquistPotential -from ..test_core import TestAbstractPotential +from ..test_core import TestAbstractPotential as AbstractPotential_Test from .test_common import MassParameterMixin, ShapeCParameterMixin class TestHernquistPotential( - TestAbstractPotential, + AbstractPotential_Test, # Parameters MassParameterMixin, ShapeCParameterMixin, diff --git a/tests/unit/potential/builtin/test_isochrone.py b/tests/unit/potential/builtin/test_isochrone.py index 5856ca6b..1b0f892b 100644 --- a/tests/unit/potential/builtin/test_isochrone.py +++ b/tests/unit/potential/builtin/test_isochrone.py @@ -5,12 +5,12 @@ import galax.potential as gp -from ..test_core import TestAbstractPotential +from ..test_core import TestAbstractPotential as AbstractPotential_Test from .test_common import MassParameterMixin, ShapeAParameterMixin class TestIsochronePotential( - TestAbstractPotential, + AbstractPotential_Test, # Parameters MassParameterMixin, ShapeAParameterMixin, diff --git a/tests/unit/potential/builtin/test_miyamotonagai.py b/tests/unit/potential/builtin/test_miyamotonagai.py index c92ad6d6..9b62aca8 100644 --- a/tests/unit/potential/builtin/test_miyamotonagai.py +++ b/tests/unit/potential/builtin/test_miyamotonagai.py @@ -5,16 +5,12 @@ import galax.potential as gp -from ..test_core import TestAbstractPotential -from .test_common import ( - MassParameterMixin, - ShapeAParameterMixin, - ShapeBParameterMixin, -) +from ..test_core import TestAbstractPotential as AbstractPotential_Test +from .test_common import MassParameterMixin, ShapeAParameterMixin, ShapeBParameterMixin class TestMiyamotoNagaiPotential( - TestAbstractPotential, + AbstractPotential_Test, # Parameters MassParameterMixin, ShapeAParameterMixin, diff --git a/tests/unit/potential/builtin/test_nfw.py b/tests/unit/potential/builtin/test_nfw.py index 082731a4..9e09e4f3 100644 --- a/tests/unit/potential/builtin/test_nfw.py +++ b/tests/unit/potential/builtin/test_nfw.py @@ -9,7 +9,7 @@ from galax.units import galactic from ..params.test_field import ParameterFieldMixin -from ..test_core import TestAbstractPotential +from ..test_core import TestAbstractPotential as AbstractPotential_Test from .test_common import MassParameterMixin @@ -47,7 +47,7 @@ def test_r_s_userfunc(self, pot_cls, fields): class TestNFWPotential( - TestAbstractPotential, + AbstractPotential_Test, # Parameters MassParameterMixin, ScaleRadiusParameterMixin, diff --git a/tests/unit/potential/builtin/test_null.py b/tests/unit/potential/builtin/test_null.py index e96fab0a..c29fba76 100644 --- a/tests/unit/potential/builtin/test_null.py +++ b/tests/unit/potential/builtin/test_null.py @@ -5,10 +5,10 @@ import galax.potential as gp -from ..test_core import TestAbstractPotential +from ..test_core import TestAbstractPotential as AbstractPotential_Test -class TestBarPotential(TestAbstractPotential): +class TestBarPotential(AbstractPotential_Test): @pytest.fixture(scope="class") def pot_cls(self) -> type[gp.NullPotential]: return gp.NullPotential diff --git a/tests/unit/potential/test_core.py b/tests/unit/potential/test_core.py index 6440b6e1..4379a9e8 100644 --- a/tests/unit/potential/test_core.py +++ b/tests/unit/potential/test_core.py @@ -11,11 +11,11 @@ from galax.units import UnitSystem, dimensionless, galactic from galax.utils import partial_jit, vectorize_method -from .test_base import TestAbstractPotentialBase +from .test_base import TestAbstractPotentialBase as AbstractPotentialBase_Test from .test_utils import FieldUnitSystemMixin -class TestAbstractPotential(TestAbstractPotentialBase, FieldUnitSystemMixin): +class TestAbstractPotential(AbstractPotentialBase_Test, FieldUnitSystemMixin): """Test the `galax.potential.AbstractPotentialBase` class.""" @pytest.fixture(scope="class")