update tests to reflect new Landscape design

tests/test_attenuation_functions.py

@@ -19,12 +19,11 @@ def test_exact_attenuation_retrieval(energy, mu):
 
 
 @pytest.mark.parametrize("energy,mu", [
-    (0.662, 0.0778),
-    (1.25, 0.06)
+    (0.662, 0.0778)
 ])
 def test_attenuation_interpolation(energy, mu):
     """
-    Test Cs-137 and Co-60 mass attenuation coefficients.
+    Test retreival for Cs-137 mass attenuation coefficient.
     """
     interp_mu = get_mass_attenuation_coeff(energy)
     assert pytest.approx(interp_mu, rel=1E-2) == mu

tests/test_fluence_rate.py

@@ -2,9 +2,7 @@ from math import dist, exp, pi
 
 import pytest
 
-from pg_rad.isotopes import CS137
-from pg_rad.landscape import Landscape
-from pg_rad.objects import PointSource
+from pg_rad.landscape import LandscapeDirector
 
 
 @pytest.fixture
@@ -27,12 +25,7 @@ def phi_ref():
 
 @pytest.fixture
 def test_landscape():
-    landscape = Landscape()
-    source = PointSource(
-        pos=(0, 0, 0),
-        activity=100E9,
-        isotope=CS137)
-    landscape.add_sources(source)
+    landscape = LandscapeDirector().build_test_landscape()
     return landscape
 
 

tests/test_sources.py

@@ -2,17 +2,17 @@ import numpy as np
 import pytest
 
 from pg_rad.objects import PointSource
-from pg_rad.isotopes import Isotope
+from pg_rad.isotopes import CS137
 
 
 @pytest.fixture
 def test_sources():
-    iso = Isotope("test", E=662, b=0)
+    iso = CS137()
     pos_a = np.random.rand(3)
     pos_b = np.random.rand(3)
 
-    a = PointSource(pos_a, activity=None, isotope=iso)
-    b = PointSource(pos_b, activity=None, isotope=iso)
+    a = PointSource(pos=pos_a, activity=None, isotope=iso)
+    b = PointSource(pos=pos_b, activity=None, isotope=iso)
 
     return pos_a, pos_b, a, b