def test_get_current_matrix(self):
potentials = cr.get_potentials(self.test_laplacian, (0, 1))
currents, triu_currents = cr.get_current_matrix(
self.test_laplacian, potentials)
calc = triu_currents.toarray()[:, 2].tolist()
ref = np.array([-1, 1, 0, 0])
self.assertTrue(np.mean(np.abs(calc - ref)) < 1e-9)
def test_get_current_through_nodes(self):
potentials = cr.get_potentials(self.test_laplacian, (0, 1))
currents, triu_currents = cr.get_current_matrix(self.test_laplacian, potentials)
triu_currents = lil_matrix(triu_currents)
node_currents = cr.get_current_through_nodes(triu_currents)
ref = np.array([1, 1, 2, 0])
self.assertTrue(np.mean(np.abs(node_currents - ref)) < 1e-9) # FAILING
def test_get_current_through_nodes(self):
potentials = cr.get_potentials(self.test_laplacian, (0, 1))
currents, triu_currents = cr.get_current_matrix(self.test_laplacian, potentials)
triu_currents = lil_matrix(triu_currents)
node_currents = cr.get_current_through_nodes(triu_currents)
ref = np.array([1, 1, 1, 0])
self.assertTrue(np.mean(np.abs(node_currents - ref)) < 1e-9)
def test_build_potentials(self):
calc = cr.get_potentials(self.test_laplacian, (0, 3))
self.assertTrue(calc[0, 0] > 1e5)
calc = cr.get_potentials(self.test_laplacian, (0, 1))
ref = np.array([[0.833333333, -0.666666666, -0.166666666666, 0.0]]).T
self.assertTrue(np.mean(np.abs(calc - ref)) < 1e-9)
def test_get_current_matrix(self):
potentials = cr.get_potentials(self.test_laplacian, (0, 1))
currents, triu_currents = cr.get_current_matrix(self.test_laplacian, potentials)
calc = triu_currents.toarray()[:, 2].tolist()
ref = np.array([2, 2, 0, 0])
self.assertTrue(np.mean(np.abs(calc - ref)) < 1e-9) # FAILING
def test_build_potentials(self):
calc = cr.get_potentials(self.test_laplacian, (0, 3))
self.assertTrue(calc[0, 0] > 1e5)
calc = cr.get_potentials(self.test_laplacian, (0, 1))
ref = np.array([[0.833333333, -0.666666666, -0.166666666666, 0.0]]).T
self.assertTrue(np.mean(np.abs(calc - ref)) < 1e-9)