free-disk/tests/data_size_to_bytes_test.py

import pytest

import free_disk


@pytest.mark.parametrize(('data_size_with_unit', 'expected_bytes'), [
    ('123', 123),
    ('123B', 123),
    ('123.0B', 123),
    ('1kB', 1000),
    ('2kB', 2000),
    ('2.5kB', 2500),
    ('2KB', 2000),
    ('8MB', 8 * (10**6)),
    ('8.5MB', 8.5 * (10**6)),
    ('32GB', 32 * (10**9)),
    ('9TB', 9 * (10**12)),
    ('3KiB', 3 * (1024**1)),
    ('40MiB', 40 * (1024**2)),
    ('512GiB', 512 * (1024**3)),
    ('7TiB', 7 * (1024**4)),
    ('123 B', 123),
    ('123\tB', 123),
    ('123.0  B', 123),
    ('1  kB', 1000),
    ('1  MiB', 1024**2),
])
def test_data_size_to_bytes(data_size_with_unit, expected_bytes):
    assert expected_bytes == free_disk.data_size_to_bytes(data_size_with_unit)


@pytest.mark.parametrize('data_size_with_unit', [
    'abcdef',
    '123G',
])
def test_data_size_to_bytes_fail(data_size_with_unit):
    with pytest.raises(ValueError):
        free_disk.data_size_to_bytes(data_size_with_unit)