def test_trace_variable(): # A random number of friends and random contacts friend_number = random.choice(range(1, 10)) friends = random.sample(range(100), friend_number) trace = generate_trace(100, 10, 1000, friends) TA_friends = analyze_trace(trace, len(friends)) assert len(TA_friends) == len(friends) assert sorted(TA_friends) == sorted(friends)
def test_trace_static(): # A fixed set and number of friends trace = generate_trace(100, 10, 1000, [1, 2, 3]) friends = analyze_trace(trace, 3) assert len(friends) == 3 assert sorted(friends) == [1, 2, 3]