def test_non_zero_entry2(self): with self.assertRaises(Exception): Triplets([1, 1, 1], [0, 1, 1]) with self.assertRaises(Exception): Triplets([1, 1, 1], [1, 0, 1]) with self.assertRaises(Exception): Triplets([1, 1, 1], [1, 1, 0])
def test_max_limit_entry2(self): with self.assertRaises(Exception): Triplets([1, 1, 1], [101, 1, 1]) with self.assertRaises(Exception): Triplets([1, 1, 1], [1, 101, 1]) with self.assertRaises(Exception): Triplets([1, 1, 1], [1, 1, 101])
def assemblage(mesh): """Assembles A and b Parameters ---------- mesh : Mesh the mesh Returns ------- (Triplets, numpy.array) matrix A and vector b """ A = Triplets() b = np.zeros((mesh.nbPoints)) for p in mesh.get_elements(2, -1): # for triangle in mesh.triangles: Dp = p.matrice_rigidite_elem() bp = p.rhs() for i in range(3): I = local_to_global(p, i) for j in range(3): J = local_to_global(p, j) A.append(I, J, Dp[i][j]) b[I] += bp[i] return (A, b)
def test_point_for_Bob(self): self.triplets = Triplets([1, 1, 1], [2, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1) self.triplets = Triplets([1, 1, 1], [1, 2, 1]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1) self.triplets = Triplets([1, 1, 1], [1, 1, 2]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1)
def test_no_point(self): self.triplets = Triplets([1, 1, 1], [1, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[0], 0) self.assertEqual(self.triplets.score[1], 0)
class TestComparation(TestTriplets): def test_point_for_Alice(self): self.triplets = Triplets([2, 1, 1], [1, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[0], 1) self.triplets = Triplets([1, 2, 1], [1, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[0], 1) self.triplets = Triplets([1, 1, 2], [1, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[0], 1) def test_point_for_Bob(self): self.triplets = Triplets([1, 1, 1], [2, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1) self.triplets = Triplets([1, 1, 1], [1, 2, 1]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1) self.triplets = Triplets([1, 1, 1], [1, 1, 2]) self.triplets.review() self.assertEqual(self.triplets.score[1], 1) def test_no_point(self): self.triplets = Triplets([1, 1, 1], [1, 1, 1]) self.triplets.review() self.assertEqual(self.triplets.score[0], 0) self.assertEqual(self.triplets.score[1], 0)
def setUp(self): self.triplets = Triplets([1, 1, 1], [1, 1, 1])
def test_correct_elements(self): self.triplets = Triplets([1, 1, 1], [1, 1, 1]) self.assertEqual(len(self.triplets.entry1), 3) self.assertEqual(len(self.triplets.entry2), 3)
def test_empty_elements(self): with self.assertRaises(Exception): Triplets([], [])
def test_entry_elements(self): #self.asserEqual(len(self.triplets.entry1) , 3) with self.assertRaises(Exception): Triplets([1, 1, 1, 1, 1], [1, 1, 1, 1, 1])