58 lines
1.2 KiB
Python
58 lines
1.2 KiB
Python
import dataclasses as dc
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from dopt_basics import system
|
|
|
|
|
|
@dc.dataclass()
|
|
class _TData:
|
|
dic: dict[str, str]
|
|
dic2: dict[str, dict[str, int]]
|
|
lst: list[int]
|
|
tup: tuple[str, ...]
|
|
string: str
|
|
id_: int
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def TData() -> _TData:
|
|
return _TData(
|
|
{"test": "test", "test2": "test"},
|
|
{"test": {"prop1": 3, "prop2": 500}},
|
|
[1, 2, 3, 4, 5, 6, 7, 8, 9],
|
|
("test", "test", "test", "test", "test", "test", "test"),
|
|
"This is one test string",
|
|
1234,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def TData_real_size(TData) -> int:
|
|
return 1435
|
|
|
|
|
|
def test_obj_size(TData, TData_real_size):
|
|
ret = system.obj_size(TData)
|
|
|
|
assert ret == TData_real_size
|
|
|
|
|
|
def test_obj_size_kb(TData, TData_real_size):
|
|
ret = system.obj_size(TData, unit="kb")
|
|
|
|
assert ret == pytest.approx(TData_real_size / 1024)
|
|
|
|
|
|
def test_obj_size_mb(TData, TData_real_size):
|
|
ret = system.obj_size(TData, unit="mb")
|
|
|
|
assert ret == pytest.approx(TData_real_size / 1024**2)
|
|
|
|
|
|
def test_obj_size_gb(TData, TData_real_size):
|
|
ret = system.obj_size(TData, unit="gb")
|
|
|
|
assert ret == pytest.approx(TData_real_size / 1024**3)
|