Newer
Older

Michał Pogoda
committed
from src.detections import DateDetection
text = "W dniu 1.01.2022 sprzedałem kota. 5 kwietnia 2021 roku kupiłem psa."

Michał Pogoda
committed
(DateDetection.AnnotationPart.TWO_DIGITS_DAY, "01"), # Only supports two digits for now
(DateDetection.AnnotationPart.OTHER, "."),
(DateDetection.AnnotationPart.TWO_DIGIT_MONTH, "01"),
(DateDetection.AnnotationPart.OTHER, "."),
(DateDetection.AnnotationPart.FOUR_DIGIT_YEAR, "2022")

Michał Pogoda
committed
(DateDetection.AnnotationPart.TWO_DIGITS_DAY, "05"), # Only supports two digits for now
(DateDetection.AnnotationPart.OTHER, " "),
(DateDetection.AnnotationPart.TEXT_MONTH, "kwietnia"),
(DateDetection.AnnotationPart.OTHER, " "),
(DateDetection.AnnotationPart.FOUR_DIGIT_YEAR, "2021"),

Michał Pogoda
committed
assert found_dates == [(7, 16, DateDetection(format_date1)), (34, 49, DateDetection(format_date2))]