You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
981 B
41 lines
981 B
2 years ago
|
import pytest
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
# from cnmodel.util.matlab_proc import MatlabProcess
|
||
|
|
import matlab.engine
|
||
|
|
|
||
|
|
|
||
|
|
def test_matlab():
|
||
|
|
global proc
|
||
|
|
try:
|
||
|
|
# proc = MatlabProcess()
|
||
|
|
proc = matlab.engine.start_matlab()
|
||
|
|
except RuntimeError:
|
||
|
|
# no matlab available; skip this test
|
||
|
|
pytest.skip("MATLAB unavailable")
|
||
|
|
|
||
|
|
base_vcount = proc.who(nargout=1) # .shape[0]
|
||
|
|
assert len(base_vcount) == 0
|
||
|
|
|
||
|
|
e4 = np.array(proc.eye(4, nargout=1))
|
||
|
|
|
||
|
|
assert isinstance(e4, np.ndarray)
|
||
|
|
assert np.all(e4 == np.eye(4))
|
||
|
|
# assert proc.who().shape[0] == base_vcount
|
||
|
|
|
||
|
|
o6_ref = np.array(proc.ones(6, nargout=1)) # _transfer=False))
|
||
|
|
# o6 = o6_ref.get()
|
||
|
|
o6 = np.array(proc.ones(6, nargout=1))
|
||
|
|
assert np.all(o6 == np.ones(6))
|
||
|
|
# print(proc.who(nargout=1))
|
||
|
|
# assert proc.who(nargout=1) == base_vcount + 1
|
||
|
|
|
||
|
|
del o6_ref
|
||
|
|
# assert proc.who().shape[0] == base_vcount
|
||
|
|
|
||
|
|
proc.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_matlab()
|