def test_add_original(self): aggregator = Aggregator() aggregator.add(Repo(True, 10, "Python", ["Flask"])) self.assertEqual(aggregator.count, 1) self.assertEqual(aggregator.original_count, 1) self.assertEqual(aggregator.forked_count, 0)
def test_aggregate(self): aggr = Aggregator() aggregated_list = aggr.aggregate(initial_networks) self.assertCountEqual(aggregated_list, result_networks)