class ArrayTestCase(unittest.TestCase):
def setUp(self):
"""this setup function will be called before each test in this class"""
self.hobj = Component()
self.hobj.add('arr1',Array(array([98.9]), iotype='in', units='ft'))
self.hobj.add('arr2', Array(array([13.2]), iotype='out', units='inch'))
self.hobj.add('arr3', Array(iotype='in', units='kg', desc='stuff'))
self.hobj.add('arr98', Array(iotype='in'))
self.hobj.add('arr99', Array(iotype='in'))
self.hobj.arr1 = [1.0, 2.0, 3.0]
self.hobj.arr2 = [[1.,2.],[3.,4.]]
self.hobj.arr3 = [1.1]
self.hobj.arr98 = [[0., 1., 0.1944, 0.1944], [0., 33., 1., 0.]];
self.hobj.arr99 = [[0, 1, 0.1944, 0.1944], [0, 0, 1, 0]]
def tearDown(self):
"""this teardown function will be called after each test"""
self.hobj = None
def test_set_to_default(self):
self.hobj.add('arr4', Array(iotype='in', units='kg'))
self.assertTrue(all(array([]) == self.hobj.arr4))
self.hobj.arr4 = [6.5]
self.assertEqual(6.5, self.hobj.arr4[0])
self.hobj.revert_to_defaults()
self.assertTrue(all(array([98.9]) == self.hobj.arr1))
self.assertTrue(all(array([]) == self.hobj.arr4))
def test_assignment(self):
# check starting value
self.assertTrue(all(array([1.,2.,3.]) == self.hobj.arr1))
# check default value
self.assertEqual([98.9], self.hobj.get_trait('arr1').trait_type.default_value)
# use convert_units to perform unit conversion
self.hobj.arr2 = convert_units(self.hobj.arr1, self.hobj.get_trait('arr1').units,
'inch')
self.assertAlmostEqual(12., self.hobj.arr2[0], 5)
self.assertAlmostEqual(24., self.hobj.arr2[1], 5)
self.assertAlmostEqual(36., self.hobj.arr2[2], 5)
def test_bogus_units(self):
try:
uf = Array([0.], iotype='in', units='bogus')
except ValueError, err:
self.assertEqual(str(err),
"Units of 'bogus' are invalid")
else:
class ArrayTestCase(unittest.TestCase):
def setUp(self):
"""this setup function will be called before each test in this class"""
self.hobj = Component()
self.hobj.add('arr1', Array(array([98.9]), iotype='in', units='ft'))
self.hobj.add('arr2', Array(array([13.2]), iotype='out', units='inch'))
self.hobj.add('arr3', Array(iotype='in', units='kg', desc='stuff'))
self.hobj.add('arr98', Array(iotype='in'))
self.hobj.add('arr99', Array(iotype='in'))
self.hobj.arr1 = [1.0, 2.0, 3.0]
self.hobj.arr2 = [[1., 2.], [3., 4.]]
self.hobj.arr3 = [1.1]
self.hobj.arr98 = [[0., 1., 0.1944, 0.1944], [0., 33., 1., 0.]]
self.hobj.arr99 = [[0, 1, 0.1944, 0.1944], [0, 0, 1, 0]]
def tearDown(self):
"""this teardown function will be called after each test"""
self.hobj = None
def test_set_to_default(self):
self.hobj.add('arr4', Array(iotype='in', units='kg'))
self.assertTrue(all(array([]) == self.hobj.arr4))
self.hobj.arr4 = [6.5]
self.assertEqual(6.5, self.hobj.arr4[0])
self.hobj.revert_to_defaults()
self.assertTrue(all(array([98.9]) == self.hobj.arr1))
self.assertTrue(all(array([]) == self.hobj.arr4))
def test_assignment(self):
# check starting value
self.assertTrue(all(array([1., 2., 3.]) == self.hobj.arr1))
# check default value
self.assertEqual([98.9],
self.hobj.get_trait('arr1').trait_type.default_value)
# use convert_units to perform unit conversion
self.hobj.arr2 = convert_units(self.hobj.arr1,
self.hobj.get_trait('arr1').units,
'inch')
self.assertAlmostEqual(12., self.hobj.arr2[0], 5)
self.assertAlmostEqual(24., self.hobj.arr2[1], 5)
self.assertAlmostEqual(36., self.hobj.arr2[2], 5)
def test_bogus_units(self):
try:
uf = Array([0.], iotype='in', units='bogus')
except ValueError, err:
self.assertEqual(str(err), "Units of 'bogus' are invalid")
else:
