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]