def test_default_control_system_import():
"""In this test we:
- assert that the lattice is indeed loaded if no execeptions are raised
- assert that the default control system is indeed cothread and that it
is loaded onto the lattice correctly
"""
assert bool(load("VMX"))
assert isinstance(load("VMX")._cs, pytac.cothread_cs.CothreadControlSystem)
def test_import_fail_raises_ControlSystemException(mock_cs_raises_ImportError):
"""In this test we:
- check that load corectly fails if cothread cannot be imported
- check that when the import of the CothreadControlSystem fails the
ImportError raised is replaced with a ControlSystemException
"""
with patch("pytac.cothread_cs.CothreadControlSystem", mock_cs_raises_ImportError):
with pytest.raises(pytac.exceptions.ControlSystemException):
load("VMX")
def pytac_lattice():
return load_csv.load("DIAD", cs.ControlSystem())
def lattice():
lat = load_csv.load("dummy", mock.MagicMock(), CURRENT_DIR_PATH / "data",
2)
return lat
def lattice():
lat = load_csv.load('dummy', mock.MagicMock(), os.path.join(CURRENT_DIR, 'data'))
return lat