refactor: update sasview api for test_sas.py

news/fix-sasmodels.rst

@@ -0,0 +1,23 @@
+**Added:**
+
+* <news item>
+
+**Changed:**
+
+* <news item>
+
+**Deprecated:**
+
+* <news item>
+
+**Removed:**
+
+* <news item>
+
+**Fixed:**
+
+* Refactored code utilizing sasmodels to use the new sasview api.
+
+**Security:**
+
+* <news item>

src/diffpy/srfit/sas/sasparser.py

@@ -22,7 +22,6 @@
 
 from diffpy.srfit.exceptions import ParseError
 from diffpy.srfit.fitbase.profileparser import ProfileParser
-from diffpy.srfit.sas.sasimport import sasimport
 
 
 class SASParser(ProfileParser):
@@ -102,12 +101,13 @@ def parseFile(self, filename):
         Raises IOError if the file cannot be read
         Raises ParseError if the file cannot be parsed
         """
+        import sasdata.dataloader.loader as sas_dataloader
 
-        Loader = sasimport("sas.dataloader.loader").Loader
+        Loader = sas_dataloader.Loader
         loader = Loader()
 
         try:
-            data = loader.load(filename)
+            data = loader.load(str(filename))
         except RuntimeError as e:
             raise ParseError(e)
         except ValueError as e:
@@ -118,7 +118,10 @@ def parseFile(self, filename):
         self._meta["filename"] = filename
         self._meta["datainfo"] = data
 
-        self._banks.append([data.x, data.y, data.dx, data.dy])
+        for data_obj in data:
+            self._banks.append(
+                [data_obj.x, data_obj.y, data_obj.dx, data_obj.dy]
+            )
         self.selectBank(0)
         return
 

src/diffpy/srfit/sas/sasprofile.py

@@ -82,7 +82,7 @@ def __init__(self, datainfo):
         datainfo
             The DataInfo object this wraps.
         """
-        self._datainfo = datainfo
+        self._datainfo = datainfo[0]
         Profile.__init__(self)
 
         self._xobs = self._datainfo.x

tests/conftest.py

@@ -7,16 +7,18 @@
 import six
 
 import diffpy.srfit.equation.literals as literals
-from diffpy.srfit.sas.sasimport import sasimport
 
 logger = logging.getLogger(__name__)
 
 
 @lru_cache()
 def has_sas():
     try:
-        sasimport("sas.pr.invertor")
-        sasimport("sas.models")
+        import sas
+        import sasmodels
+
+        del sas
+        del sasmodels
         return True
     except ImportError:
         return False

tests/test_characteristicfunctions.py

@@ -18,9 +18,11 @@
 
 import numpy
 import pytest
+from sasmodels.sasview_model import find_model, load_standard_models
 
 import diffpy.srfit.pdf.characteristicfunctions as cf
-from diffpy.srfit.sas.sasimport import sasimport
+
+load_standard_models()
 
 # # Global variables to be assigned in setUp
 # cf = None
@@ -34,7 +36,7 @@ def testSphere(sas_available):
         pytest.skip("sas package not available")
     radius = 25
     # Calculate sphere cf from SphereModel
-    SphereModel = sasimport("sas.models.SphereModel").SphereModel
+    SphereModel = find_model("sphere")
     model = SphereModel()
     model.setParam("radius", radius)
     ff = cf.SASCF("sphere", model)
@@ -56,10 +58,10 @@ def testSpheroid(sas_available):
     prad = 20.9
     erad = 33.114
     # Calculate cf from EllipsoidModel
-    EllipsoidModel = sasimport("sas.models.EllipsoidModel").EllipsoidModel
+    EllipsoidModel = find_model("ellipsoid")
     model = EllipsoidModel()
-    model.setParam("radius_a", prad)
-    model.setParam("radius_b", erad)
+    model.setParam("radius_polar", prad)
+    model.setParam("radius_equatorial", erad)
     ff = cf.SASCF("spheroid", model)
     r = numpy.arange(0, 100, 1 / numpy.pi, dtype=float)
     fr1 = ff(r)
@@ -79,7 +81,7 @@ def testShell(sas_available):
     radius = 19.2
     thickness = 7.8
     # Calculate cf from VesicleModel
-    VesicleModel = sasimport("sas.models.VesicleModel").VesicleModel
+    VesicleModel = find_model("vesicle")
     model = VesicleModel()
     model.setParam("radius", radius)
     model.setParam("thickness", thickness)
@@ -103,7 +105,7 @@ def testCylinder(sas_available):
     radius = 100
     length = 30
 
-    CylinderModel = sasimport("sas.models.CylinderModel").CylinderModel
+    CylinderModel = find_model("cylinder")
     model = CylinderModel()
     model.setParam("radius", radius)
     model.setParam("length", length)

tests/test_sas.py

@@ -17,8 +17,12 @@
 import numpy
 import pytest
 
+# Use the updated SasView model API to load models
+from sasmodels.sasview_model import find_model, load_standard_models
+
 from diffpy.srfit.sas import SASGenerator, SASParser, SASProfile
-from diffpy.srfit.sas.sasimport import sasimport
+
+load_standard_models()
 
 # ----------------------------------------------------------------------------
 # FIXME: adjust sensitivity of the pytest.approx statements when ready to test
@@ -113,7 +117,7 @@ def testParser(sas_available, datafile):
 def test_generator(sas_available):
     if not sas_available:
         pytest.skip("sas package not available")
-    SphereModel = sasimport("sas.models.SphereModel").SphereModel
+    SphereModel = find_model("sphere")
     model = SphereModel()
     gen = SASGenerator("sphere", model)
     for pname in model.params:
@@ -140,25 +144,27 @@ def test_generator(sas_available):
 def testGenerator2(sas_available, datafile):
     if not sas_available:
         pytest.skip("sas package not available")
-    EllipsoidModel = sasimport("sas.models.EllipsoidModel").EllipsoidModel
+    EllipsoidModel = find_model("ellipsoid")
     model = EllipsoidModel()
     gen = SASGenerator("ellipsoid", model)
 
     # Load the data using SAS tools
-    Loader = sasimport("sas.dataloader.loader").Loader
+    import sasdata.dataloader.loader as sas_dataloader
+
+    Loader = sas_dataloader.Loader
     loader = Loader()
     data = datafile("sas_ellipsoid_testdata.txt")
-    datainfo = loader.load(data)
+    datainfo = loader.load(str(data))
     profile = SASProfile(datainfo)
 
     gen.setProfile(profile)
     gen.scale.value = 1.0
-    gen.radius_a.value = 20
-    gen.radius_b.value = 400
+    gen.radius_polar.value = 20
+    gen.radius_equatorial.value = 400
     gen.background.value = 0.01
 
     y = gen(profile.xobs)
     diff = profile.yobs - y
     res = numpy.dot(diff, diff)
-    assert 0 == pytest.approx(res)
+    assert 0 == pytest.approx(res, abs=1e-3)
     return