start plot tests

Signed-off-by: Nathaniel Starkman (@nstarman) <[email protected]>

Author: nstarman

pyproject.toml

@@ -15,7 +15,7 @@ force_grid_wrap = 0
 use_parentheses = "True"
 ensure_newline_before_comments = "True"
 sections = ["FUTURE", "STDLIB", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"]
-known_third_party = ["astropy", "extension_helpers", "galpy", "numpy", "pytest", "scipy", "setuptools"]
+known_third_party = ["astropy", "extension_helpers", "galpy", "matplotlib", "numpy", "pytest", "scipy", "setuptools"]
 known_localfolder = "sample_scf"
 
 import_heading_stdlib = "BUILT-IN"

sample_scf/conftest.py

@@ -16,7 +16,8 @@
 # THIRD PARTY
 import numpy as np
 import pytest
-from galpy.potential import SCFPotential
+from galpy.df import isotropicHernquistdf
+from galpy.potential import HernquistPotential, SCFPotential
 
 try:
     # THIRD PARTY
@@ -74,17 +75,18 @@ def pytest_configure(config):
 # ============================================================================
 # Fixtures
 
+# Hernquist
+_Acos = np.zeros((5, 6, 6))
+_Acos_hern = _Acos.copy()
+_Acos_hern[0, 0, 0] = 1
+_hernquist_potential = SCFPotential(Acos=_Acos_hern)
+hernquist_df = isotropicHernquistdf(HernquistPotential())
+
 
 @pytest.fixture(autouse=True, scope="session")
 def hernquist_scf_potential():
     """Make a SCF of a Hernquist potential."""
-    Acos = np.zeros((5, 6, 6))
-
-    Acos_hern = Acos.copy()
-    Acos_hern[0, 0, 0] = 1
-
-    hernpot = SCFPotential(Acos=Acos_hern)
-    return hernpot
+    return _hernquist_potential
 
 
 # /def
@@ -109,9 +111,6 @@ def hernquist_scf_potential():
 )
 def potentials(request):
     if request.param in ("hernquist_scf_potential", "other_hernquist_scf_potential"):
-        Acos = np.zeros((5, 6, 6))
-        Acos_hern = Acos.copy()
-        Acos_hern[0, 0, 0] = 1
-        potential = SCFPotential(Acos=Acos_hern)
+        potential = _hernquist_potential
 
     yield potential

sample_scf/tests/test_base.py

@@ -54,6 +54,9 @@ def setup_class(self):
         self.cls = rvtestsampler
         self.cls_args = ()
 
+        self.cdf_args = (0,)
+        self.cdf_kwargs = {}
+
         self.cdf_time_scale = 3e-6
         self.rvs_time_scale = 1e-4
 
@@ -71,6 +74,13 @@ def sampler(self, potentials):
     # ===============================================================
     # Method Tests
 
+    @pytest.mark.skip("TODO")
+    def test_cdf(self, sampler, expected):
+        """Test :meth:`sample_scf.base.rv_potential.cdf`."""
+        assert False
+
+    # /def
+
     @pytest.mark.parametrize(
         "size, random, expected",
         [
@@ -119,9 +129,6 @@ def test_rvs_time_scaling(self, sampler, size):
 
     # /def
 
-    # ===============================================================
-    # Image tests
-
 
 # /class
 
@@ -162,6 +169,7 @@ def sampler(self, potentials):
     # /def
 
     # ===============================================================
+    # Method Tests
 
     def test_rsampler(self, sampler):
         """Test :meth:`sample_scf.base.SCFSamplerBase.rsampler`."""