def test_connect_users_in_a_graph(self):
graph = Graph()
user1 = User()
user2 = User()
graph.add_user(user1)
graph.add_user(user2)
graph.connect_users(user1.get_id(), user2.get_id())
assert user2 in user1.get_connections() and user1 in user2.get_connections()
def construct_graph_of_count(count):
graph = Graph()
user1 = User()
graph.add_user(user1)
for i in range(0, count-1):
user = User()
graph.add_user(user)
graph.connect_users(user1.get_id(), user.get_id())
return graph
def test_add_users_to_a_graph(self):
graph = Graph()
user = User()
user_id = user.get_id()
graph.add_user(user)
assert user_id in graph.users and graph.users[user_id] == user
def test_user_id_generation(self):
user1 = User()
user2 = User()
assert user2.get_id() == user1.get_id() + 1
