Rooba
/
rich
mirror of https://github.com/Textualize/rich.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify the download column test

This commit is contained in:
amartya-dev 2020-10-04 23:00:52 +05:30
parent 82c2f44cd2
commit d1cd347a31
1 changed files with 4 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_progress.py
View File

@ -82,23 +82,10 @@ def test_time_remaining_column():
def test_download_progress_uses_decimal_units() -> None:
column = DownloadColumn()
size_params = [pow(10, ex) for ex in range(3, 25, 3)]
units = ["KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB"]
# Check rendering params for all sizes
for i in range(len(size_params)):
test_task = Task(
i, f"test_{units[i]}", size_params[i], 0, _get_time=lambda: 1.0
)
list_rendered = []
for n in range(10):
test_task.completed = size_params[i] / 10 * (n + 1)
list_rendered.append(str(column.render(test_task)))
expected = [f"0.{completed}/1.0 {units[i]}" for completed in range(1, 10)] + [
f"1.0/1.0 {units[i]}"
]
assert list_rendered == expected
test_task = Task(1, "test", 1000, 500, _get_time=lambda: 1.0)
rendered_progress = str(column.render(test_task))
expected = "0.5/1.0 KB"
assert rendered_progress == expected
def test_task_ids():