def setUp(self):
self.mesh = dolfin.UnitCube(1,1,1)
self.V = dolfin.FunctionSpace(self.mesh, "Nedelec 1st kind H(curl)", 1)
self.DUT = FillamentSource(self.V)
self.fillament_current = 2
self.fillament_endpoints = np.array([[0,0,0.7], [0,0,0.2]])
self.source_parameters = dict(
I=self.fillament_current, endpoints=self.fillament_endpoints)
class test_fillament_source(unittest.TestCase):
def setUp(self):
self.mesh = dolfin.UnitCube(1,1,1)
self.V = dolfin.FunctionSpace(self.mesh, "Nedelec 1st kind H(curl)", 1)
self.DUT = FillamentSource(self.V)
self.fillament_current = 2
self.fillament_endpoints = np.array([[0,0,0.7], [0,0,0.2]])
self.source_parameters = dict(
I=self.fillament_current, endpoints=self.fillament_endpoints)
def test_set_source_parameters(self):
self.DUT.set_source_parameters(self.source_parameters)
self.assertTrue(np.all(self.DUT.direction == [0,0,-1]))
self.assertAlmostEqual(self.DUT.length, 0.5)
def test_get_current_source(self):
self.DUT.set_source_parameters(self.source_parameters)
cs = self.DUT.get_current_source()
self.assertTrue(isinstance(cs, FillamentCurrentSource))
self.assertEqual(cs.value, self.fillament_current)
self.assertTrue(np.all(cs.source_endpoints == self.fillament_endpoints))
