def manual_initialization(n_nodes):
nodes = GaussLobattoNodes()
nodes.init(n_nodes)
assert_equal(nodes.num_nodes, n_nodes,
"Number of nodes should be set")
assert_is_instance(nodes.nodes, np.ndarray,
"Nodes should be a numpy.ndarray")
assert_equal(nodes.nodes.size, n_nodes,
"There should be correct number of nodes")
class GaussLobattoNodesTest(unittest.TestCase):
def setUp(self):
self._test_obj = GaussLobattoNodes()
def test_default_initialization(self):
self.assertIsNone(self._test_obj.num_nodes,
"Number of nodes should be initialized as 'None'")
self.assertIsNone(self._test_obj.nodes,
"Nodes list should be initializes as 'None'")
def test_correctness_of_selected_nodes(self):
self._test_obj.init(3)
self.assertAlmostEqual(self._test_obj.nodes[0], -1.0)
self.assertAlmostEqual(self._test_obj.nodes[1], 0.0)
self.assertAlmostEqual(self._test_obj.nodes[2], 1.0)
self.setUp()
self._test_obj.init(4)
self.assertAlmostEqual(self._test_obj.nodes[0], -1.0)
self.assertAlmostEqual(self._test_obj.nodes[1], -np.sqrt(1.0 / 5.0))
self.assertAlmostEqual(self._test_obj.nodes[2], np.sqrt(1.0 / 5.0))
self.assertAlmostEqual(self._test_obj.nodes[3], 1.0)
self.setUp()
self._test_obj.init(5)
self.assertAlmostEqual(self._test_obj.nodes[0], -1.0)
self.assertAlmostEqual(self._test_obj.nodes[1], -np.sqrt(3.0 / 7.0))
self.assertAlmostEqual(self._test_obj.nodes[2], 0.0)
self.assertAlmostEqual(self._test_obj.nodes[3], np.sqrt(3.0 / 7.0))
self.assertAlmostEqual(self._test_obj.nodes[4], 1.0)