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
