From 7f7bcd5d8fab629f78930f4a173e3bf7fe14ae92 Mon Sep 17 00:00:00 2001 From: doctorpangloss <2229300+doctorpangloss@users.noreply.github.com> Date: Tue, 4 Nov 2025 18:33:31 -0800 Subject: [PATCH] Fix tests (no rerelease necessary) --- tests/unit/test_operator_nodes.py | 3 ++- tests/unit/test_sdpa.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/unit/test_operator_nodes.py b/tests/unit/test_operator_nodes.py index 3c563e620..b9a5cdad3 100644 --- a/tests/unit/test_operator_nodes.py +++ b/tests/unit/test_operator_nodes.py @@ -122,7 +122,8 @@ def test_binary_operation_lazy_check(): # Test standard ops assert n.check_lazy_status(op="eq") == ["lhs", "rhs"] - assert n.check_lazy_status(op="add", lhs=1, rhs=2) == ["lhs", "rhs"] + assert n.check_lazy_status(op="add", lhs=1, rhs=2) == [] + assert n.check_lazy_status(op="add", lhs=None, rhs=None) == ["lhs", "rhs"] # Test 'and' assert n.check_lazy_status(op="and", lhs=None) == ["lhs"] diff --git a/tests/unit/test_sdpa.py b/tests/unit/test_sdpa.py index 926160c39..5daf0715d 100644 --- a/tests/unit/test_sdpa.py +++ b/tests/unit/test_sdpa.py @@ -91,7 +91,7 @@ def test_sdpa_import_exception(): importlib.reload(comfy.ops) assert comfy.ops.scaled_dot_product_attention is comfy.ops._scaled_dot_product_attention - mock_logger.debug.assert_called_once() + mock_logger.debug.assert_called() # Check that the log message contains the exception info assert "Could not set sdpa backend priority." in mock_logger.debug.call_args[0][0]