Add tests for time

2025-07-17 21:22:08 +02:00
parent 77bb515aa4
commit eae08bbf7b
2 changed files with 217 additions and 19 deletions
@@ -60,7 +60,7 @@ class TimeToUnix(ComfyNodeABC):
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "dt": (IO.DATETIME, {}),
+                "datetime": (IO.DATETIME, {}),
            }
        }

@@ -70,11 +70,11 @@ class TimeToUnix(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "to_unix"

-    def to_unix(self, dt: datetime.datetime) -> tuple[float]:
+    def to_unix(self, datetime: datetime.datetime) -> tuple[float]:
        """
        Converts the given datetime object to a Unix timestamp.
        """
-        return (dt.timestamp(),)
+        return (datetime.timestamp(),)


 class UnixToTime(ComfyNodeABC):
@@ -111,7 +111,7 @@ class TimeFormat(ComfyNodeABC):
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "dt": (IO.DATETIME, {}),
+                "datetime": (IO.DATETIME, {}),
                "format_string": (IO.STRING, {"default": "%Y-%m-%d %H:%M:%S"}),
            }
        }
@@ -122,11 +122,11 @@ class TimeFormat(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "format_time"

-    def format_time(self, dt: datetime.datetime, format_string: str) -> tuple[str]:
+    def format_time(self, datetime: datetime.datetime, format_string: str) -> tuple[str]:
        """
        Formats a datetime object into a string.
        """
-        return (dt.strftime(format_string),)
+        return (datetime.strftime(format_string),)


 class TimeParse(ComfyNodeABC):
@@ -193,7 +193,7 @@ class TimeAddDelta(ComfyNodeABC):
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "dt": (IO.DATETIME, {}),
+                "datetime": (IO.DATETIME, {}),
                "delta": (IO.TIMEDELTA, {}),
            }
        }
@@ -203,11 +203,11 @@ class TimeAddDelta(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "add"

-    def add(self, dt: datetime.datetime, delta: datetime.timedelta) -> tuple[datetime.datetime]:
+    def add(self, datetime: datetime.datetime, delta: datetime.timedelta) -> tuple[datetime.datetime]:
        """
        Adds a timedelta to a datetime object.
        """
-        return (dt + delta,)
+        return (datetime + delta,)


 class TimeSubtractDelta(ComfyNodeABC):
@@ -218,7 +218,7 @@ class TimeSubtractDelta(ComfyNodeABC):
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "dt": (IO.DATETIME, {}),
+                "datetime": (IO.DATETIME, {}),
                "delta": (IO.TIMEDELTA, {}),
            }
        }
@@ -228,11 +228,11 @@ class TimeSubtractDelta(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "subtract"

-    def subtract(self, dt: datetime.datetime, delta: datetime.timedelta) -> tuple[datetime.datetime]:
+    def subtract(self, datetime: datetime.datetime, delta: datetime.timedelta) -> tuple[datetime.datetime]:
        """
        Subtracts a timedelta from a datetime object.
        """
-        return (dt - delta,)
+        return (datetime - delta,)


 class TimeDifference(ComfyNodeABC):
@@ -243,8 +243,8 @@ class TimeDifference(ComfyNodeABC):
    def INPUT_TYPES(cls):
        return {
            "required": {
-                "dt1": (IO.DATETIME, {}),
-                "dt2": (IO.DATETIME, {}),
+                "datetime1": (IO.DATETIME, {}),
+                "datetime2": (IO.DATETIME, {}),
            }
        }

@@ -253,11 +253,11 @@ class TimeDifference(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "difference"

-    def difference(self, dt1: datetime.datetime, dt2: datetime.datetime) -> tuple[datetime.timedelta]:
+    def difference(self, datetime1: datetime.datetime, datetime2: datetime.datetime) -> tuple[datetime.timedelta]:
        """
        Calculates the duration between two datetime objects.
        """
-        return (dt1 - dt2,)
+        return (datetime1 - datetime2,)


 class TimeExtract(ComfyNodeABC):
@@ -268,7 +268,7 @@ class TimeExtract(ComfyNodeABC):
    @classmethod
    def INPUT_TYPES(cls):
        return {
-            "required": { "dt": (IO.DATETIME, {}) }
+            "required": { "datetime": (IO.DATETIME, {}) }
        }

    RETURN_TYPES = (IO.INT, IO.INT, IO.INT, IO.INT, IO.INT, IO.INT, IO.INT, IO.INT)
@@ -277,11 +277,12 @@ class TimeExtract(ComfyNodeABC):
    DESCRIPTION = cleandoc(__doc__ or "")
    FUNCTION = "extract"

-    def extract(self, dt: datetime.datetime) -> tuple[int, int, int, int, int, int, int, int]:
+    def extract(self, datetime: datetime.datetime) -> tuple[int, int, int, int, int, int, int, int]:
        """
        Extracts all time components from a datetime object.
        """
-        return (dt.year, dt.month, dt.day, dt.hour, dt.minute, dt.second, dt.microsecond, dt.weekday())
+        return (datetime.year, datetime.month, datetime.day, datetime.hour,
+                datetime.minute, datetime.second, datetime.microsecond, datetime.weekday())


 NODE_CLASS_MAPPINGS = {
@@ -0,0 +1,197 @@
+#import pytest
+
+from datetime import datetime, timedelta
+from src.basic_data_handling.time_nodes import (
+    TimeNow,
+    TimeToUnix,
+    UnixToTime,
+    TimeFormat,
+    TimeParse,
+    TimeDelta,
+    TimeAddDelta,
+    TimeSubtractDelta,
+    TimeDifference,
+    TimeExtract,
+)
+
+def test_time_now():
+    node = TimeNow()
+    result = node.get_now()
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], datetime)
+
+    # Test with trigger parameter
+    result_with_trigger = node.get_now(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)
+    result = node.to_unix(test_datetime)
+
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], float)
+    # 2023-01-01 12:00:00 should convert to a specific timestamp
+    expected_timestamp = test_datetime.timestamp()
+    assert result[0] == expected_timestamp
+
+def test_unix_to_time():
+    node = UnixToTime()
+    # Test with a specific timestamp (2023-01-01 12:00:00)
+    test_timestamp = 1672574400.0  # This is 2023-01-01 12:00:00 in UTC
+    result = node.from_unix(test_timestamp)
+
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], datetime)
+
+    # The returned datetime will be in local timezone, so hours might differ
+    # Just check that the date is correct and hours are reasonable
+    returned_dt = result[0]
+    assert returned_dt.year == 2023
+    assert returned_dt.month == 1
+    assert returned_dt.day == 1
+    # Allow for timezone differences (typically within ±12 hours)
+    assert 0 <= returned_dt.hour < 24
+    assert returned_dt.minute == 0
+    assert returned_dt.second == 0
+
+def test_time_format():
+    node = TimeFormat()
+    test_datetime = datetime(2023, 1, 1, 12, 30, 45)
+
+    # Test with default format
+    result = node.format_time(test_datetime, "%Y-%m-%d %H:%M:%S")
+    assert result == ("2023-01-01 12:30:45",)
+
+    # Test with custom format
+    result = node.format_time(test_datetime, "%d/%m/%Y")
+    assert result == ("01/01/2023",)
+
+    # Test with time-only format
+    result = node.format_time(test_datetime, "%H:%M")
+    assert result == ("12:30",)
+
+def test_time_parse():
+    node = TimeParse()
+
+    # Test standard format
+    result = node.parse_time("2023-01-01 12:30:45", "%Y-%m-%d %H:%M:%S")
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], datetime)
+    assert result[0] == datetime(2023, 1, 1, 12, 30, 45)
+
+    # Test custom format
+    result = node.parse_time("01/01/2023", "%d/%m/%Y")
+    assert result[0] == datetime(2023, 1, 1, 0, 0, 0)
+
+    # Test time-only format
+    result = node.parse_time("12:30", "%H:%M")
+    parsed_datetime = result[0]
+    assert parsed_datetime.hour == 12
+    assert parsed_datetime.minute == 30
+
+def test_time_delta():
+    node = TimeDelta()
+
+    # Test with days only
+    result = node.create_delta(days=5)
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], timedelta)
+    assert result[0].days == 5
+
+    # Test with multiple parameters
+    result = node.create_delta(days=1, hours=2, minutes=30)
+    delta = result[0]
+    # Convert to seconds for easier comparison
+    total_seconds = delta.total_seconds()
+    expected_seconds = 24*60*60 + 2*60*60 + 30*60  # 1 day, 2 hours, 30 minutes
+    assert total_seconds == expected_seconds
+
+    # Test with all parameters
+    result = node.create_delta(
+        days=1,
+        seconds=30,
+        microseconds=500,
+        milliseconds=100,
+        minutes=5,
+        hours=2,
+        weeks=1
+    )
+    delta = result[0]
+    # A week plus a day
+    assert delta.days == 8
+    # Convert the remainder to seconds for comparison
+    remainder_seconds = delta.seconds
+    expected_remainder = 2*60*60 + 5*60 + 30 + 0.1  # 2 hours, 5 minutes, 30 seconds, 100 milliseconds
+    assert abs(remainder_seconds - expected_remainder) < 1
+    assert delta.microseconds == 100500
+
+def test_time_add_delta():
+    node = TimeAddDelta()
+    test_datetime = datetime(2023, 1, 1, 12, 0, 0)
+    test_delta = timedelta(days=1, hours=2, minutes=30)
+
+    result = node.add(test_datetime, test_delta)
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], datetime)
+
+    expected_datetime = datetime(2023, 1, 2, 14, 30, 0)  # 1 day, 2 hours, 30 minutes later
+    assert result[0] == expected_datetime
+
+def test_time_subtract_delta():
+    node = TimeSubtractDelta()
+    test_datetime = datetime(2023, 1, 2, 14, 30, 0)
+    test_delta = timedelta(days=1, hours=2, minutes=30)
+
+    result = node.subtract(test_datetime, test_delta)
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], datetime)
+
+    expected_datetime = datetime(2023, 1, 1, 12, 0, 0)  # 1 day, 2 hours, 30 minutes earlier
+    assert result[0] == expected_datetime
+
+def test_time_difference():
+    node = TimeDifference()
+    datetime1 = datetime(2023, 1, 2, 14, 30, 0)
+    datetime2 = datetime(2023, 1, 1, 12, 0, 0)
+
+    result = node.difference(datetime1, datetime2)
+    assert isinstance(result, tuple)
+    assert len(result) == 1
+    assert isinstance(result[0], timedelta)
+
+    expected_delta = timedelta(days=1, hours=2, minutes=30)
+    assert result[0] == expected_delta
+
+    # Test reverse order (should give negative timedelta)
+    result = node.difference(datetime2, datetime1)
+    assert result[0] == -expected_delta
+
+def test_time_extract():
+    node = TimeExtract()
+    test_datetime = datetime(2023, 1, 2, 14, 30, 45, 123456)
+
+    result = node.extract(test_datetime)
+    assert isinstance(result, tuple)
+    assert len(result) == 8
+
+    year, month, day, hour, minute, second, microsecond, weekday = result
+
+    assert year == 2023
+    assert month == 1
+    assert day == 2
+    assert hour == 14
+    assert minute == 30
+    assert second == 45
+    assert microsecond == 123456
+    # January 2, 2023 was a Monday (weekday 0)
+    assert weekday == 0