test core

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

Author: nstarman

sample_scf/tests/test_base.py

@@ -11,6 +11,7 @@
 import astropy.units as u
 import numpy as np
 import pytest
+from astropy.utils.misc import NumpyRNGContext
 from numpy.testing import assert_allclose
 
 # LOCAL
@@ -57,11 +58,17 @@ def setup_class(self):
             (None, 0, 0.5488135039273248),
             (1, 2, 0.43599490214200376),
             ((3, 1), 4, (0.9670298390136767, 0.5472322491757223, 0.9726843599648843)),
+            ((3, 1), None, (0.9670298390136767, 0.5472322491757223, 0.9726843599648843)),
         ],
     )
     def test_rvs(self, size, random, expected):
-        """Test :meth:`sample_scf.base.rv_continuous_modrvs.rvs`."""
-        assert_allclose(self.sampler.rvs(size=size, random_state=random), expected, atol=1e-16)
+        """Test :meth:`sample_scf.base.rv_continuous_modrvs.rvs`.
+
+        The ``NumpyRNGContext`` is to control the random generator used to make
+        the RandomState. For ``random != None``, this doesn't matter.
+        """
+        with NumpyRNGContext(4):
+            assert_allclose(self.sampler.rvs(size=size, random_state=random), expected, atol=1e-16)
 
     # /def
 
@@ -156,6 +163,12 @@ def test_rvs(self, sampler, id, size, random):
 
     # /def
 
+    # ===============================================================
+    # Time Scaling Tests
+
+    # ===============================================================
+    # Image tests
+
 
 # /class
 

sample_scf/tests/test_core.py

@@ -6,9 +6,6 @@
 ##############################################################################
 # IMPORTS
 
-# THIRD PARTY
-import pytest
-
 # LOCAL
 from .test_base import Test_SCFSamplerBase
 from sample_scf import core
@@ -18,7 +15,6 @@
 ##############################################################################
 
 
-@pytest.mark.skip("TODO!")
 class Test_SCFSampler(Test_SCFSamplerBase):
     """Test :class:`sample_scf.core.SCFSample`."""
 

sample_scf/tests/test_init.py

@@ -22,8 +22,16 @@ def test_expected_imports():
     """Test can import expected modules and objects."""
     # LOCAL
     import sample_scf
+    from sample_scf import core, sample_exact, sample_intrp
 
     assert inspect.ismodule(sample_scf)
+    assert inspect.ismodule(core)
+    assert inspect.ismodule(sample_exact)
+    assert inspect.ismodule(sample_intrp)
+
+    assert sample_scf.SCFSampler is core.SCFSampler
+    assert sample_scf.SCFSamplerExact is sample_exact.SCFSampler
+    assert sample_scf.SCFSamplerInterp is sample_intrp.SCFSampler
 
 
 # /def