def test__grand_total_one_row(self) -> None: # Arrange data = copy.deepcopy(SAMPLE_DATA_ONE_ROW) # Act output = pypistats._grand_total(data) # Assert assert output == SAMPLE_DATA_ONE_ROW
def test__grand_total_recent(self) -> None: # Arrange data = copy.deepcopy(SAMPLE_DATA_RECENT) # Act output = pypistats._grand_total(data) # Assert assert output == SAMPLE_DATA_RECENT
def test__grand_total_recent(self): # Arrange data = copy.deepcopy(SAMPLE_DATA_RECENT) # Act output = pypistats._grand_total(data) # Assert self.assertEqual(output, SAMPLE_DATA_RECENT)
def test__grand_total_one_row(self): # Arrange data = copy.deepcopy(SAMPLE_DATA_ONE_ROW) # Act output = pypistats._grand_total(data) # Assert self.assertEqual(output, SAMPLE_DATA_ONE_ROW)
def test__grand_total(self) -> None: # Arrange data = copy.deepcopy(PYTHON_MINOR_DATA) original_len = len(data) # Act output = pypistats._grand_total(data) # Assert assert len(output) == original_len + 1 assert output[-1]["category"] == "Total" assert output[-1]["downloads"] == 9_355_317
def test__grand_total(self): # Arrange data = copy.deepcopy(PYTHON_MINOR_DATA) original_len = len(data) # Act output = pypistats._grand_total(data) # Assert self.assertEqual(len(output), original_len + 1) self.assertEqual(output[-1]["category"], "Total") self.assertEqual(output[-1]["downloads"], 9_355_317)
def test__colourify(self) -> None: # Arrange data = [ { "category": "2.7", "downloads": 1 }, { "category": "3.5", "downloads": 10 }, { "category": "3.10", "downloads": 89 }, ] expected_output = [ # red { "category": "2.7", "downloads": 1, "percent": "\x1b[31m1.00%\x1b[0m" }, # yellow { "category": "3.5", "downloads": 10, "percent": "\x1b[33m10.00%\x1b[0m" }, # green { "category": "3.10", "downloads": 89, "percent": "\x1b[32m89.00%\x1b[0m" }, { "category": "Total", "downloads": 100 }, ] data = pypistats._percent(data) data = pypistats._grand_total(data) # Act output = pypistats._colourify(data) # Assert assert output == expected_output