def test_should_get_pair_from_database_if_exist(self): programmer1 = Programmer(name='Minkey', id=1) programmer2 = Programmer(name='Minnie', id=2) programmer1.save() programmer2.save() programmers = [programmer1,programmer2] pair_times = 2 Pair(first=programmer2, second=programmer1, count =pair_times).save() pairs = create_pairs(programmers) self.assertEqual(pair_times, pairs[0].count)
def test_should_create_pairs(self): programmers = [Programmer(name="Minkey", id=1), Programmer(name="Minne", id=2)] pairs = create_pairs(programmers) self.assertEqual(1, len(pairs))