test: update tests for SimpleSZ

galsci · Jul 23, 2024 · c1e6d30 · c1e6d30
1 parent c59ee7a
commit c1e6d30
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/pysm3/models/__init__.py b/pysm3/models/__init__.py
@@ -15,7 +15,6 @@
     SPT_CIB_map_scaling,
     WebSkyCMB,
     WebSkyCIB,
-    WebSkySZ,
-    WebSkyRadioGalaxies,
-    AgoraSZ
+    SimpleSZ,
+    WebSkyRadioGalaxies
 )
diff --git a/pysm3/tests/test_websky.py b/pysm3/tests/test_websky.py
@@ -12,7 +12,7 @@
 from .. import (
     SPT_CIB_map_scaling,
     WebSkyCIB,
-    WebSkySZ,
+    SimpleSZ,
     WebSkyRadioGalaxies,
 )  # , WebSkyCMBTensor
 
@@ -126,7 +126,7 @@ def test_sz(tmp_path, monkeypatch, sz_type):
         test_map *= 1e-6
     hp.write_map(path / filename, test_map)
 
-    tsz = WebSkySZ(nside, "0.4", sz_type=sz_type)
+    tsz = SimpleSZ(nside, template_name=str(path/filename), sz_type=sz_type, max_nside=8192)
 
     freq = 100 * u.GHz
     tsz_map = tsz.get_emission(freq).to(u.uK_CMB, equivalencies=u.cmb_equivalencies(freq))