102 lines
2.5 KiB
Python
102 lines
2.5 KiB
Python
"""Tests for utility functions."""
|
|
|
|
from devtoolbelt.utils import (
|
|
format_bytes,
|
|
format_duration,
|
|
parse_key_value,
|
|
pluralize,
|
|
truncate,
|
|
)
|
|
|
|
|
|
class TestFormatBytes:
|
|
"""Tests for format_bytes function."""
|
|
|
|
def test_bytes(self):
|
|
assert format_bytes(500) == "500.00 B"
|
|
|
|
def test_kilobytes(self):
|
|
assert format_bytes(2048) == "2.00 KB"
|
|
|
|
def test_megabytes(self):
|
|
assert format_bytes(1048576) == "1.00 MB"
|
|
|
|
def test_gigabytes(self):
|
|
assert format_bytes(1073741824) == "1.00 GB"
|
|
|
|
|
|
class TestFormatDuration:
|
|
"""Tests for format_duration function."""
|
|
|
|
def test_milliseconds(self):
|
|
assert format_duration(0.5) == "500ms"
|
|
|
|
def test_seconds(self):
|
|
assert format_duration(30) == "30.0s"
|
|
|
|
def test_minutes(self):
|
|
assert format_duration(120) == "2.0m"
|
|
|
|
def test_hours(self):
|
|
assert format_duration(7200) == "2.0h"
|
|
|
|
def test_days(self):
|
|
assert format_duration(172800) == "2.0d"
|
|
|
|
|
|
class TestParseKeyValue:
|
|
"""Tests for parse_key_value function."""
|
|
|
|
def test_empty_string(self):
|
|
assert parse_key_value("") == {}
|
|
|
|
def test_single_pair(self):
|
|
result = parse_key_value("key=value")
|
|
assert result == {"key": "value"}
|
|
|
|
def test_multiple_pairs(self):
|
|
result = parse_key_value("key1=value1,key2=value2,key3=value3")
|
|
assert result == {
|
|
"key1": "value1",
|
|
"key2": "value2",
|
|
"key3": "value3"
|
|
}
|
|
|
|
def test_values_with_spaces(self):
|
|
result = parse_key_value("name=John Doe,role=Admin")
|
|
assert result == {"name": "John Doe", "role": "Admin"}
|
|
|
|
def test_missing_value(self):
|
|
result = parse_key_value("key")
|
|
assert result == {}
|
|
|
|
|
|
class TestPluralize:
|
|
"""Tests for pluralize function."""
|
|
|
|
def test_singular(self):
|
|
assert pluralize(1, "item") == "item"
|
|
|
|
def test_plural(self):
|
|
assert pluralize(5, "item") == "items"
|
|
|
|
def test_custom_plural(self):
|
|
assert pluralize(1, "child", "children") == "child"
|
|
assert pluralize(3, "child", "children") == "children"
|
|
|
|
|
|
class TestTruncate:
|
|
"""Tests for truncate function."""
|
|
|
|
def test_no_truncation_needed(self):
|
|
assert truncate("hello", 10) == "hello"
|
|
|
|
def test_truncation(self):
|
|
assert truncate("hello world", 8) == "hello..."
|
|
|
|
def test_custom_suffix(self):
|
|
assert truncate("hello world", 8, suffix="***") == "hello***"
|
|
|
|
def test_exact_length(self):
|
|
assert truncate("hello", 5) == "hello"
|