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