def test_user_connect(): """Test the graph connection behavior.""" user1 = User(1) user2 = User(2) user1.connect(user2) assert user2 in user1.adjacent assert user1 in user2.adjacent
def test_infection_model(): """Make sure users pull version from their infection.""" user = User(1, 1) assert user.version == 1 infection = Infection(2) user.infection = infection infection.add(user) assert user.version == 2