mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-08-29 04:26:20 +00:00
18f1d55d77
* test(normalize): make optional tests deterministic Signed-off-by: aoright <102943475+aoright@users.noreply.github.com> * docs(changelog): add entry for #881 / #860 normalize test determinism fixes --------- Signed-off-by: aoright <102943475+aoright@users.noreply.github.com> Co-authored-by: Mohd Kaif <98801504+KaifAhmad1@users.noreply.github.com> Co-authored-by: KaifAhmad1 <kaifahmad087@gmail.com>
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
import unittest
|
|
from datetime import datetime, timedelta, timezone
|
|
from unittest.mock import patch
|
|
|
|
from semantica.normalize.date_normalizer import (
|
|
DateNormalizer,
|
|
RelativeDateProcessor,
|
|
TimeZoneNormalizer,
|
|
)
|
|
|
|
|
|
class TestDateNormalizer(unittest.TestCase):
|
|
def setUp(self):
|
|
self.normalizer = DateNormalizer()
|
|
|
|
def test_normalize_date_iso(self):
|
|
# Test ISO8601 parsing
|
|
self.assertEqual(
|
|
self.normalizer.normalize_date("2023-01-01", format="date"), "2023-01-01"
|
|
)
|
|
self.assertEqual(
|
|
self.normalizer.normalize_date("2023-01-01T12:00:00", format="ISO8601"),
|
|
"2023-01-01T12:00:00+00:00",
|
|
)
|
|
|
|
def test_normalize_date_relative(self):
|
|
class FixedDateTime(datetime):
|
|
@classmethod
|
|
def now(cls, tz=None):
|
|
fixed = cls(2024, 1, 2, 0, 30)
|
|
return fixed if tz is None else fixed.replace(tzinfo=tz)
|
|
|
|
with patch("semantica.normalize.date_normalizer.datetime", FixedDateTime):
|
|
self.assertEqual(
|
|
self.normalizer.normalize_date("today", format="date"),
|
|
"2024-01-02",
|
|
)
|
|
|
|
def test_normalize_timezone(self):
|
|
# Test timezone conversion
|
|
# "2023-01-01T12:00:00+01:00" -> UTC should be "2023-01-01T11:00:00+00:00"
|
|
normalized = self.normalizer.normalize_date(
|
|
"2023-01-01T12:00:00+01:00", timezone="UTC"
|
|
)
|
|
self.assertEqual(normalized, "2023-01-01T11:00:00+00:00")
|
|
|
|
def test_parse_temporal_expression(self):
|
|
# Test range parsing
|
|
result = self.normalizer.parse_temporal_expression(
|
|
"from 2023-01-01 to 2023-01-31"
|
|
)
|
|
self.assertIsNotNone(result.get("range"))
|
|
|
|
|
|
class TestTimeZoneNormalizer(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tz_normalizer = TimeZoneNormalizer()
|
|
|
|
def test_normalize_timezone_obj(self):
|
|
dt = datetime(2023, 1, 1, 12, 0, 0)
|
|
# Assuming default is UTC if not specified or naive
|
|
normalized = self.tz_normalizer.normalize_timezone(dt, "UTC")
|
|
# Check offset instead of object identity
|
|
self.assertEqual(
|
|
normalized.tzinfo.utcoffset(normalized), timezone.utc.utcoffset(None)
|
|
)
|
|
|
|
|
|
class TestRelativeDateProcessor(unittest.TestCase):
|
|
def setUp(self):
|
|
self.processor = RelativeDateProcessor()
|
|
|
|
def test_process_relative_expression(self):
|
|
# "3 days ago"
|
|
dt = self.processor.process_relative_expression("3 days ago")
|
|
self.assertIsInstance(dt, datetime)
|
|
# Roughly check delta
|
|
# Use datetime.now() since result is naive
|
|
diff = datetime.now() - dt
|
|
self.assertTrue(timedelta(days=2, hours=23) < diff < timedelta(days=3, hours=1))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|