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)