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: