Add test for PSpecBeam base class

HERA-Team · Apr 24, 2018 · 67d464b · 67d464b
1 parent 39c4d9e
commit 67d464b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/hera_pspec/tests/test_pspecbeam.py b/hera_pspec/tests/test_pspecbeam.py
@@ -129,4 +129,12 @@ def test_Gaussbeam(self):
         # test taper execution
         scalar = self.gauss.compute_pspec_scalar(lower_freq, upper_freq, num_freqs, num_steps=5000, taper='blackman')
         self.assertAlmostEqual(scalar / 22123832163.072491, 1.0, delta=1e-8)
-
+
+    def test_PSpecBeamBase(self):
+        """
+        Test that base class can be instantiated.
+        """
+        bm1 = pspecbeam.PSpecBeamBase()
+
+        # Check that user-defined cosmology can be specified
+        bm2 = pspecbeam.PSpecBeamBase(cosmo=conversions.Cosmo_Conversions())