Tetst+save_outputs.py: for COM, allow opening the editor; wrap faulthandler.

Without the editor, we cannot get the report filename sometimes

Work around faulthandler when running pytest and fix some COM test issues.

Author: PMeira

tests/save_outputs.py

@@ -346,7 +346,7 @@ def get_archive_fn(live_fn):
         import comtypes.client
         DSS = comtypes.client.CreateObject("OpenDSSEngine.DSS")
         DSS = dss.patch_dss_com(DSS)
-        DSS.Text.Command = r'set editor=ignore_me_invalid_executable'
+        #DSS.Text.Command = r'set editor=ignore_me_invalid_executable' -- need to let it open for some reports :|
         print("Using official OpenDSS COM:", DSS.Version)
         com_ver = DSS.Version.split(' ')[1]
         suffix = f'-COM-{platform.machine()}-{com_ver}'

tests/test_batch.py

@@ -1,16 +1,24 @@
 import os, io
+try:
+    from ._settings import BASE_DIR, WIN32
+except ImportError:
+    from _settings import BASE_DIR, WIN32
 # import numpy as np
 import pandas as pd
+if WIN32:
+    # When running pytest, the faulthandler seems to eager to grab FPC's exceptions, even when handled
+    import faulthandler
+    faulthandler.disable()
+    import dss
+    faulthandler.enable()
+else:
+    import dss
+
 from dss import dss, Edit, IDSS
 from dss.IObj import (
     Vsource, Transformer, LineCode, Load, Line, Capacitor,
     Connection as Conn, RegControl, DimensionUnits as Units,
 )
-try:
-    from ._settings import BASE_DIR
-except ImportError:
-    from _settings import BASE_DIR
-
 LoadModel = Load.LoadModel
 
 loads_cols = 'name,bus1,phases,conn,model,kV,kW,kvar'.split(',')

tests/test_ctrlqueue.py

@@ -6,9 +6,9 @@
 import numpy as np
 
 try:
-    from ._settings import BASE_DIR
+    from ._settings import BASE_DIR, WIN32
 except ImportError:
-    from _settings import BASE_DIR
+    from _settings import BASE_DIR, WIN32
 
 
 USE_COM = False # Change to True to test with the COM DLL
@@ -17,7 +17,15 @@ def test_ctrlqueue():
     '''Example of implementing a simple voltage control for Capacitors via the interface'''
 
     if not USE_COM:
-        from dss import DSS as DSSobj
+        if WIN32:
+            # When running pytest, the faulthandler seems to eager to grab FPC's exceptions, even when handled
+            import faulthandler
+            faulthandler.disable()
+            from dss import DSS as DSSobj
+            faulthandler.enable()
+        else:
+            from dss import DSS as DSSobj
+        
     else:
         import os
         old_cd = os.getcwd()

tests/test_general.py

@@ -3,15 +3,24 @@
 #       covers detailed check of API states, etc.
 import sys, os, itertools, threading
 from time import perf_counter
-import dss
-from dss import DSS, IDSS, DSSException, SparseSolverOptions, SolveModes, set_case_insensitive_attributes, DSSCompatFlags
-import numpy as np
-import pytest
 try:
     from ._settings import BASE_DIR, ZIP_FN, WIN32
 except ImportError:
     from _settings import BASE_DIR, ZIP_FN, WIN32
 
+if WIN32:
+    # When running pytest, the faulthandler seems to eager to grab FPC's exceptions, even when handled
+    import faulthandler
+    faulthandler.disable()
+    import dss
+    faulthandler.enable()
+else:
+    import dss
+
+from dss import DSS, IDSS, DSSException, SparseSolverOptions, SolveModes, set_case_insensitive_attributes, DSSCompatFlags
+import numpy as np
+import pytest
+
 
 def setup_function():
     DSS.ClearAll()
@@ -589,10 +598,10 @@ def test_essentials(DSS: IDSS = DSS):
     Text.Command = 'new line.line1 bus=1 bus2=2'
     Text.Command = 'new line.line2 bus=2 bus2=3'
     assert DSS.ActiveCircuit.Name == 'test789'
-    assert len(DSS.ActiveCircuit.Buses) == 0
+    assert len(DSS.ActiveCircuit.ActiveBus) == 0
     Text.Command = 'MakeBusList'
-    assert len(DSS.ActiveCircuit.Buses) == 4
-    for expected, b in zip(['sourcebus', '1', '2', '3'], DSS.ActiveCircuit.Buses):
+    assert len(DSS.ActiveCircuit.ActiveBus) == 4
+    for expected, b in zip(['sourcebus', '1', '2', '3'], DSS.ActiveCircuit.ActiveBus):
         assert expected == b.Name
 
 
@@ -605,7 +614,8 @@ def test_patch_comtypes():
 def test_patch_win32com():
     if WIN32:
         import win32com.client
-        DSS_COM = dss.patch_dss_com(win32com.client.Dispatch("OpenDSSengine.DSS"))
+        win32com.client.Dispatch("OpenDSSengine.DSS")
+        DSS_COM = dss.patch_dss_com(win32com.client.gencache.EnsureDispatch("OpenDSSengine.DSS"))
         test_essentials(DSS_COM)
 
 if __name__ == '__main__':

tests/test_obj.py

@@ -1,13 +1,24 @@
 import numpy as np
+try:
+    from ._settings import BASE_DIR, WIN32
+except ImportError:
+    from _settings import BASE_DIR, WIN32
+
+if WIN32:
+    # When running pytest, the faulthandler seems to eager to grab FPC's exceptions, even when handled
+    import faulthandler
+    faulthandler.disable()
+    import dss
+    faulthandler.enable()
+else:
+    import dss
+
 from dss import dss, Edit, IDSS #, YMatrixModes
 from dss.IObj import (
     Vsource, Transformer, LineCode, Load, Line, Capacitor, 
     Connection as Conn, RegControl, DimensionUnits as Units,
 )    
-try:
-    from ._settings import BASE_DIR
-except ImportError:
-    from _settings import BASE_DIR
+
 
 
 LoadModel = Load.LoadModel

tests/test_past_issues.py

@@ -5,15 +5,21 @@
 import numpy as np
 import pytest
 
-WIN32 = (sys.platform == 'win32')
-DSS.AllowEditor = False
-if os.path.exists('../../electricdss-tst/'):
-    BASE_DIR = os.path.abspath('../../electricdss-tst/')
-    ZIP_FN = os.path.abspath('data/13Bus.zip')
+try:
+    from ._settings import BASE_DIR, WIN32, ZIP_FN
+except ImportError:
+    from _settings import BASE_DIR, WIN32, ZIP_FN
+
+if WIN32:
+    # When running pytest, the faulthandler seems to eager to grab FPC's exceptions, even when handled
+    import faulthandler
+    faulthandler.disable()
+    import dss
+    faulthandler.enable()
 else:
-    BASE_DIR = os.path.abspath('../electricdss-tst/')
-    ZIP_FN = os.path.abspath('tests/data/13Bus.zip')
+    import dss
 
+DSS.AllowEditor = False
 
 def test_rxmatrix():
     DSS.ClearAll()