Example #1
0
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
Example #2
0
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