tests.pocketutils.tools.test_unit_tools

Complexity

Total Complexity

4

Size/Duplication

Total Lines	69
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	TestUnitTools.test_approx_time_wrt()	0	27	1
A	TestUnitTools.test_delta_time_to_str()	0	8	1
A	TestUnitTools.test_ms_to_minsec()	0	7	1
A	TestUnitTools.test_round_to_sigfigs()	0	6	1

# SPDX-FileCopyrightText: Copyright 2020-2023, Contributors to pocketutils
# SPDX-PackageHomePage: https://github.com/dmyersturnbull/pocketutils
# SPDX-License-Identifier: Apache-2.0
from datetime import datetime
from typing import Self

import pytest
from pocketutils.core.chars import Chars
from pocketutils.tools.unit_tools import UnitTools


class TestUnitTools:
    def test_delta_time_to_str(self: Self) -> None:
        f = UnitTools.pretty_timedelta
        assert f(15) == "15s"
        assert f(313) == "5.22min"
        assert f(15, space=Chars.narrownbsp) == "15" + Chars.narrownbsp + "s"
        assert f(15 * 60) == "15min"
        assert f(15 * 60 + 5) == "15.08min"
        assert f(15 * 60 * 60 + 5 * 60) == "15.08hr"

    def test_approx_time_wrt(self: Self) -> None:
        f = UnitTools.approx_time_wrt
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 2)) == "2021-01-02 00:00"
        assert f(datetime(2021, 1, 1), datetime(1996, 1, 1)) == "1996"
        assert f(datetime(2021, 1, 1), datetime(2021, 5, 5)) == "2021-05"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 5)) == "2021-01-05"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 6)) == "2021-01-01 06:00"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 6, 22)) == "2021-01-01 06:22"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 2, 6, 22)) == "2021-01-02 06:22"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 2, 2, 22)) == "2021-01-02 02:22"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 2, 2, 22, 40)) == "2021-01-02 02:22"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 0, 1, 40)) == "2021-01-01 00:01:40"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 0, 0, 0, 222222)) == "2021-01-01 00:00:00.222"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 0, 0, 0, 222)) == "2021-01-01 00:00:00.000222"
        assert f(datetime(2021, 1, 1), datetime(2021, 1, 1, 0, 0, 0, 0)) == "2021-01-01 00:00:00.000000"
        assert f(datetime(2021, 1, 10), datetime(2021, 1, 5)) == "2021-01-05"  # negative delta
        assert f(datetime(2021, 9, 30, 23), datetime(2021, 10, 1)) == "2021-10-01 00:00"
        assert f(datetime(2021, 10, 1), datetime(2021, 9, 30, 23)) == "2021-09-30 23:00"
        assert f(datetime(2021, 9, 27, 1), datetime(2021, 10, 1)) == "2021-10-01"
        assert f(datetime(2021, 10, 1), datetime(2021, 9, 25)) == "2021-09-25"
        assert (
            f(
                datetime(2021, 1, 1),
                datetime(2021, 1, 1, 0, 0, 0, 0),
                no_date_if_today=True,
            )
            == "00:00:00.000000"
        )

    def test_ms_to_minsec(self: Self) -> None:
        f = UnitTools.milliseconds_to_min_sec
        assert f(15) == "15ms"
        assert f(15 * 1000) == "00:15"
        assert f(15 * 60 * 1000) == "15:00"
        assert f(15 * 60 * 60 * 1000) == "15:00:00"
        assert f(15 * 24 * 60 * 60 * 1000) == "15d:00:00:00"

    def test_round_to_sigfigs(self: Self) -> None:
        f = UnitTools.round_to_sigfigs
        assert str(f(0.0012, 3)) == "0.0012"
        assert str(f(0.0012, 1)) == "0.001"
        assert str(f(0.0012 / 1000 / 1000, 3)) == "1.2e-09"
        assert str(f(0.0012 / 1000 / 1000, 1)) == "1e-09"


if __name__ == "__main__":
    pytest.main()