def test_has_group_False(self):
b = Board(2, 3)
b.all_positions = [
[Ball(TypeBall.simple, 'green'), Ball(TypeBall.simple, 'green')],
[Ball(TypeBall.simple, 'orange'), Ball(TypeBall.simple, 'orange')],
[Ball(TypeBall.simple, 'yellow'), Ball(TypeBall.simple, 'green')]
]
self.assertFalse(b.has_group())
def test_set_table_with_group(self):
b = Board(2, 3)
b.all_positions = [
[Ball(TypeBall.simple, 'green'), Ball(TypeBall.simple, 'green')],
[Ball(TypeBall.simple, 'orange'), Ball(TypeBall.simple, 'orange')],
[Ball(TypeBall.simple, 'yellow'), Ball(TypeBall.simple, 'green')]
]
b.set_table_with_group()
self.assertTrue(b.has_group())
def test_for_board_2_3(self):
b = Board(2, 3)
b.all_positions = [
[Ball(TypeBall.simple, 'green'), Ball(TypeBall.simple, 'green')],
[Ball(TypeBall.simple, 'orange'), Ball(TypeBall.simple, 'orange')],
[Ball(TypeBall.simple, 'yellow'), Ball(TypeBall.simple, 'green')]
]
group = b.group_for(0, 0)
answer = {(0, 0), (0, 1)}
self.assertEqual(group, answer)
def test_kill_the_group_for_without_None(self):
b = Board(2, 3)
b.all_positions = [
[Ball(TypeBall.simple, 'green'), Ball(TypeBall.simple, 'green')],
[Ball(TypeBall.simple, 'orange'), Ball(TypeBall.simple, 'green')],
[Ball(TypeBall.simple, 'yellow'), Ball(TypeBall.simple, 'green')]
]
b.kill_the_group_for(0, 1)
count_None = 0
for row in b.all_positions:
for elem in row:
if elem is None:
count_None += 1
self.assertEqual(count_None, 0)
