def test_get_last_partners_ids(self): self.assertEqual( frozenset(Talk.get_last_partners_ids(self.stranger_0)), frozenset([ self.stranger_1.id, self.stranger_2.id, self.stranger_3.id, ]), ) self.assertEqual( frozenset(Talk.get_last_partners_ids(self.stranger_2)), frozenset([ self.stranger_0.id, self.stranger_3.id, self.stranger_4.id, ]), )