def test_get_length(self): # print MaxCover.length_of_total_elements(world1) # 37 self.assertTrue(len(MaxCover.create_universe(world1)) == MaxCover.length_of_total_elements(world1)) # This is 34 print MaxCover.length_of_total_elements(selection1)
def test_findFriendEnemy(self): X = [[1, 2],[2, 3, 4],[3,4,5,8],[5,6,7],[7,8]] m = MaxCover() friend, enemy = MaxCover.find_friend_enemy(X, [1,2]) self.assertTrue(same(friend, [[3,4,5,8],[5,6,7],[7,8]])) self.assertTrue(same(enemy, [[2,3,4]])) friend, enemy = m.find_friend_enemy(X, [2,3,4]) self.assertTrue(same(friend, [[5,6,7],[7,8]])) self.assertTrue(same(enemy, [[1,2],[3,4,5,8]])) friend, enemy = m.find_friend_enemy(X, [3,4,5,8]) self.assertTrue(same(friend, [[1,2]])) self.assertTrue(same(enemy, [[5,6,7],[7,8],[2,3,4]])) friend, enemy = m.find_friend_enemy(X, [5,6,7]) self.assertTrue(same(friend, [[1,2],[2,3,4]])) self.assertTrue(same(enemy, [[3,4,5,8],[7,8]])) friend, enemy = m.find_friend_enemy(X, [7,8]) self.assertTrue(same(friend, [[1,2],[2,3,4]])) self.assertTrue(same(enemy, [[3,4,5,8],[5,6,7]]))