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