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])