def test_asdict(self):
aggregator = Aggregator()
expected = {
"public_repo_count": 0,
"public_repo_breakdown": {
"original_repo_count": 0,
"forked_repo_count": 0
},
"watcher_count": 0,
"languages": [],
"topics": []
}
self.assertEqual(aggregator.asdict(), expected)
expected2 = {
"public_repo_count": 1,
"public_repo_breakdown": {
"original_repo_count": 1,
"forked_repo_count": 0
},
"watcher_count": 10,
"languages": [{
"name": "python",
"count": 1
}],
"topics": [{
"name": "flask",
"count": 1
}]
}
aggregator.add(Repo(True, 10, "python", ["flask"]))
self.assertEqual(aggregator.asdict(), expected2)
def test_add_languages(self):
aggregator = Aggregator()
aggregator.add(Repo(False, 10, "Python", ["Flask"]))
self.assertEqual(aggregator.languages, {"python": 1})
aggregator.add(Repo(False, 10, "python", ["Flask"]))
self.assertEqual(aggregator.languages, {"python": 2})
aggregator.add(Repo(False, 10, "java", ["Flask"]))
self.assertEqual(aggregator.languages, {"python": 2, "java": 1})
aggregator.add(Repo(False, 10, None, ["Flask"]))
self.assertEqual(aggregator.languages, {"python": 2, "java": 1})
def test_add_topics(self):
aggregator = Aggregator()
aggregator.add(Repo(False, 10, "python", ["Flask"]))
self.assertEqual(aggregator.topics, {"flask": 1})
aggregator.add(Repo(False, 10, "python", ["flask"]))
self.assertEqual(aggregator.topics, {"flask": 2})
aggregator.add(Repo(False, 10, "python", ["flask", "numpy"]))
self.assertEqual(aggregator.topics, {"flask": 3, "numpy": 1})
def test_add_none(self):
aggregator = Aggregator()
aggregator.add(Repo(None, 10, "Python", ["Flask"]))
self.assertEqual(aggregator.count, 1)
self.assertEqual(aggregator.original_count, 0)
self.assertEqual(aggregator.forked_count, 0)