from verbrauch import start_new_year, convert_line_to_consumpt

EPSILON = 0.001

def test_start_new_year():
    line = "# 2025\n"
    (is_new_year, year_nr) = start_new_year(line)
    assert is_new_year
    assert year_nr == "2025"

def test_not_start_new_year():
    line = "[2025]\n"
    (is_new_year, year_nr) = start_new_year(line)
    assert not is_new_year
    assert year_nr is None

def test_not_start_new_year():
    line = " # 2025\n"
    (is_new_year, year_nr) = start_new_year(line)
    print(is_new_year, year_nr)
    assert is_new_year
    assert year_nr == "2025"

def test_convert_line_to_consumpt():
    line = " 5,5;   30;   32; 33,2; 38; 22; 21; 35,3"
    consum = convert_line_to_consumpt(line)
    expected_values = [5.5, 30, 32, 33.2, 38, 22, 21, 35.3]
    print(consum)
    for i in range(len(consum)):
        value = consum[i]
        expected = expected_values[i]
        assert abs(value - expected) < EPSILON