diff --git a/tests/test_time_nodes.py b/tests/test_time_nodes.py index a810a17..63c14ae 100644 --- a/tests/test_time_nodes.py +++ b/tests/test_time_nodes.py @@ -3,6 +3,7 @@ from datetime import datetime, timedelta from src.basic_data_handling.time_nodes import ( TimeNow, + TimeNowUTC, TimeToUnix, UnixToTime, TimeFormat, @@ -27,6 +28,19 @@ def test_time_now(): assert len(result_with_trigger) == 1 assert isinstance(result_with_trigger[0], datetime) +def test_time_now_utc(): + node = TimeNowUTC() + result = node.get_now_utc() + assert isinstance(result, tuple) + assert len(result) == 1 + assert isinstance(result[0], datetime) + + # Test with trigger parameter + result_with_trigger = node.get_now_utc(trigger="any value") + assert isinstance(result_with_trigger, tuple) + assert len(result_with_trigger) == 1 + assert isinstance(result_with_trigger[0], datetime) + def test_time_to_unix(): node = TimeToUnix() test_datetime = datetime(2023, 1, 1, 12, 0, 0)