From c1e6d3080dd74349c600aa2421333355709514fb Mon Sep 17 00:00:00 2001
From: Andrea Zonca <code@andreazonca.com>
Date: Mon, 22 Jul 2024 23:06:01 -0500
Subject: [PATCH] test: update tests for SimpleSZ

---
 pysm3/models/__init__.py   | 5 ++---
 pysm3/tests/test_websky.py | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/pysm3/models/__init__.py b/pysm3/models/__init__.py
index ff1d8668..e9630fb9 100644
--- 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
index 388e91e0..630008c2 100644
--- 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))