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)