from src.detectors.date.pl import detect_dates_pl
from src.entity_types import EntityTypes


def test_detect_dates_pl():
    text = "W dniu 1.01.2022 sprzedałem kota. 5 kwietnia 2021 roku kupiłem psa."
    found_dates = detect_dates_pl(text)

    assert found_dates == [(7, 16, EntityTypes.DATE), (34, 49, EntityTypes.DATE)]