Update test_unit_MSX.py

OpenWaterAnalytics · Apr 16, 2024 · f958eaf · f958eaf
1 parent 427f4bf
commit f958eaf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/epyt/tests/test_unit_MSX.py b/epyt/tests/test_unit_MSX.py
@@ -1,5 +1,4 @@
-from epyt.epanet import epanet, epanetmsxapi
-from epyt import networks
+from epyt import epanet, networks
 import unittest
 import os
 
@@ -256,7 +255,7 @@ def test_MSXgetqual(self):
         inpname = os.path.join(DIRNAME, 'msx-examples', 'net2-cl2.inp')
         self.epanetClass = epanet(inpname, msx=True)
         file_path = os.path.join(DIRNAME, 'msx-examples', 'net2-cl2.msx')
-        self.msxClass = epanetmsxapi(file_path)
+        self.msxClass = self.epanetClass.loadMSXFile(file_path)
 
         self.msxClass.MSXsolveH()
         self.msxClass.MSXsolveQ()
@@ -281,4 +280,4 @@ def test_MSXgetqual(self):
 
 
 if __name__ == "__main__":
-    unittest.main()
+    unittest.main()