def test_get_preferred_matlab(self):
# cannot rely on the output, but must not crash
print matlab_proxy.get_preferred_matlab()
class TestMatlabWrapper(unittest.TestCase):
def test_unknowns(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'stat2.m'))
def default(obj):
if isinstance(obj, FileRef):
return repr(obj)
if isinstance(obj, numpy.ndarray):
return repr(obj)
raise TypeError(repr(obj) + " is not JSON serializable")
# print(json.dumps({'params': c._init_params_dict, 'unknowns': c._init_unknowns_dict}, default=default))
self.assertIn('a', c._init_params_dict)
self.assertIn('m', c._init_unknowns_dict)
@unittest.skipUnless(matlab_proxy.get_preferred_matlab(),
"Fails with PythonEngine")
def test_values(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'stat2.m'))
unknowns = {}
c.solve_nonlinear({'a': 2.5, 'b': 3.5, 'c': 4.0}, unknowns, {})
# print(repr(unknowns))
self.assertEqual(unknowns['m'], 35)
self.assertEqual(unknowns['s'], [[2.5, 3.5, 4.0]])
self.assertEqual(unknowns['w']['m'], unknowns['m'])
self.assertEqual(unknowns['strarra'], ['Mercury', 'Venus', 'Earth'])
def test_single_output(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'single_value.m'))
unknowns = {}
c.solve_nonlinear({
'x': numpy.float64(2),
'y': numpy.float64(5)
}, unknowns, {})
print(repr(unknowns))
self.assertEqual(unknowns['z'], 7)
@unittest.skipUnless(matlab_proxy.get_preferred_matlab(
), "Fails with PythonEngine: need smop support for `MException` and `error`"
)
def test_error(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'test_error.m'))
def default(obj):
if isinstance(obj, FileRef):
return repr(obj)
if isinstance(obj, numpy.ndarray):
return repr(obj)
raise TypeError(repr(obj) + " is not JSON serializable")
unknowns = {}
c.solve_nonlinear({'mode': 0}, unknowns, {})
with self.assertRaises(openmdao.api.AnalysisError):
c.solve_nonlinear({'mode': 1}, unknowns, {})
with self.assertRaises(openmdao.api.AnalysisError):
c.solve_nonlinear({'mode': 2}, unknowns, {})
def test_get_preferred_matlab(self):
# cannot rely on the output, but must not crash
print matlab_proxy.get_preferred_matlab()
class TestBareMatlabWrapper(unittest.TestCase):
def test_unknowns(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'bare_file.m'))
def default(obj):
if isinstance(obj, FileRef):
return repr(obj)
if isinstance(obj, numpy.ndarray):
return repr(obj)
raise TypeError(repr(obj) + " is not JSON serializable")
# import json
# print(json.dumps({'params': c._init_params_dict, 'unknowns': c._init_unknowns_dict}, default=default))
self.assertIn('input1', c._init_params_dict)
self.assertIn('output2', c._init_unknowns_dict)
@unittest.skipUnless(matlab_proxy.get_preferred_matlab(),
"Fails with PythonEngine")
def test_values(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'bare_file.m'))
unknowns = {}
c.solve_nonlinear(
{
'input1': 2.5,
'input2': 3.5,
'input3': numpy.array([4.5, 9]),
'input4': 'asdf',
'input5': ['asdffa', 'asdff'],
'input6': {
'x': 2,
'y': 4
}
}, unknowns, {})
# print(repr(unknowns))
self.assertEqual(unknowns['output1'], 5)
self.assertEqual(numpy.ndarray, type(unknowns['output3']))
self.assertEqual(
unknowns['output3'].tolist(),
[[9, 18]
]) # TODO: Do we want to reshape 1xn arrays to be 1-dimensional?
self.assertEqual(unknowns['output4'], 'asdf')
self.assertEqual(unknowns['output5'], ['asdffa', 'asdff'])
self.assertEqual(unknowns['output6']['x'], 4)
def test_values_simple(self):
c = MatlabWrapper(
os.path.join(os.path.dirname(os.path.abspath(__file__)),
'bare_file_simple.m'))
unknowns = {}
c.solve_nonlinear(
{
'input1': 2.5,
'input2': 3.5,
'input3': 4,
'input4': 'asdf',
'input6': 3
}, unknowns, {})
# print(repr(unknowns))
self.assertEqual(unknowns['output1'], 5)
self.assertEqual(unknowns['output3'], 8)
self.assertEqual(unknowns['output4'], 'asdf')
self.assertEqual(unknowns['output6'], 2)