import pytest from src.basic_data_handling.comparison_nodes import ( Equal, NotEqual, GreaterThan, LessThan, GreaterThanOrEqual, LessThanOrEqual, StringComparison, NumberInRange, IsNull, CompareLength, ) def test_equal(): node = Equal() assert node.compare(10, 10) == (True,) assert node.compare(10, 20) == (False,) assert node.compare("hello", "hello") == (True,) assert node.compare([1, 2], [1, 2]) == (True,) assert node.compare(None, None) == (True,) # Test complex objects assert node.compare({"a": 1}, {"a": 1}) == (True,) assert node.compare({"a": 1}, {"a": 2}) == (False,) def test_not_equal(): node = NotEqual() assert node.compare(10, 10) == (False,) assert node.compare(10, 20) == (True,) assert node.compare("hello", "world") == (True,) assert node.compare([1, 2], [1, 3]) == (True,) assert node.compare(None, []) == (True,) # Test complex objects assert node.compare({"a": 1}, {"b": 1}) == (True,) assert node.compare({"a": 1}, {"a": 1}) == (False,) def test_greater_than(): node = GreaterThan() assert node.compare(10, 5) == (True,) assert node.compare(5, 10) == (False,) assert node.compare(10, 10) == (False,) assert node.compare(5.6, 2.3) == (True,) # Test negative numbers assert node.compare(-5, -10) == (True,) assert node.compare(-10, -5) == (False,) # Test zeroes assert node.compare(0, -1) == (True,) assert node.compare(0, 0) == (False,) def test_less_than(): node = LessThan() assert node.compare(5, 10) == (True,) assert node.compare(10, 5) == (False,) assert node.compare(10, 10) == (False,) assert node.compare(2.3, 5.6) == (True,) # Test negative numbers assert node.compare(-10, -5) == (True,) assert node.compare(-5, -10) == (False,) # Test zeroes assert node.compare(-1, 0) == (True,) assert node.compare(0, 0) == (False,) def test_greater_than_or_equal(): node = GreaterThanOrEqual() assert node.compare(10, 10) == (True,) assert node.compare(15, 10) == (True,) assert node.compare(5, 10) == (False,) assert node.compare(5.5, 5.5) == (True,) # Test negative numbers assert node.compare(-5, -10) == (True,) assert node.compare(-5, -5) == (True,) # Test zeroes assert node.compare(0, 0) == (True,) assert node.compare(0.0, -0.0) == (True,) def test_less_than_or_equal(): node = LessThanOrEqual() assert node.compare(10, 10) == (True,) assert node.compare(5, 15) == (True,) assert node.compare(15, 5) == (False,) assert node.compare(3.1, 3.1) == (True,) # Test negative numbers assert node.compare(-10, -5) == (True,) assert node.compare(-5, -5) == (True,) # Test zeroes assert node.compare(0, 0) == (True,) assert node.compare(-0.0, 0.0) == (True,) def test_string_comparison(): node = StringComparison() # Case-sensitive tests assert node.compare("hello", "hello", "==", "True") == (True,) assert node.compare("apple", "banana", "<", "True") == (True,) assert node.compare("apple", "banana", ">", "True") == (False,) assert node.compare("apple", "apple", "!=", "True") == (False,) assert node.compare("banana", "apple", ">=", "True") == (True,) assert node.compare("apple", "banana", "<=", "True") == (True,) # Case-insensitive tests assert node.compare("Hello", "hello", "==", "False") == (True,) assert node.compare("Apple", "BANANA", "<", "False") == (True,) assert node.compare("Grape", "GRAPE", ">=", "False") == (True,) assert node.compare("ZEBRA", "zoo", "<", "False") == (True,) assert node.compare("ZOO", "zebra", ">", "False") == (True,) # Empty string tests assert node.compare("", "hello", "<", "True") == (True,) assert node.compare("hello", "", ">", "True") == (True,) assert node.compare("", "", "==", "True") == (True,) def test_number_in_range(): node = NumberInRange() # Inclusive range tests assert node.check_range(50, 0, 100) == (True,) assert node.check_range(0, 0, 100) == (True,) assert node.check_range(100, 0, 100) == (True,) # Exclusive range tests assert node.check_range(0, 0, 100, include_min="False") == (False,) assert node.check_range(100, 0, 100, include_max="False") == (False,) assert node.check_range(50, 0, 100, include_min="False", include_max="False") == (True,) # Out of range tests assert node.check_range(-1, 0, 100) == (False,) assert node.check_range(101, 0, 100) == (False,) # Edge cases assert node.check_range(0, 0, 0) == (True,) assert node.check_range(0, 0, 0, include_min="False") == (False,) assert node.check_range(0, 0, 0, include_max="False") == (False,) # Negative range tests assert node.check_range(-50, -100, -10) == (True,) assert node.check_range(-10, -100, -10) == (True,) assert node.check_range(-100, -100, -10) == (True,) def test_is_null(): node = IsNull() assert node.check_null(None) == (True,) assert node.check_null(0) == (False,) assert node.check_null("") == (False,) assert node.check_null([]) == (False,) assert node.check_null(False) == (False,) # Additional edge cases assert node.check_null([None]) == (False,) # List containing None assert node.check_null("None") == (False,) # String "None" def test_compare_length(): node = CompareLength() # Valid comparisons with different operators assert node.compare_length([1, 2, 3], "==", 3) == (True, 3) assert node.compare_length("hello", "!=", 4) == (True, 5) assert node.compare_length((1, 2), ">", 1) == (True, 2) assert node.compare_length([1, 2, 3], "<", 4) == (True, 3) assert node.compare_length("test", ">=", 4) == (True, 4) assert node.compare_length([1], "<=", 1) == (True, 1) # Different container types assert node.compare_length({"a", "b", "c"}, "==", 3) == (True, 3) # Set assert node.compare_length({"key1": 1, "key2": 2}, "==", 2) == (True, 2) # Dict # Empty containers assert node.compare_length([], "==", 0) == (True, 0) assert node.compare_length("", "==", 0) == (True, 0) assert node.compare_length({}, "==", 0) == (True, 0) # Invalid comparisons assert node.compare_length({"key": "value"}, "<", 1) == (False, 1) # Dict with 1 item assert node.compare_length(123, "==", 0) == (False, -1) # Non-container input assert node.compare_length(None, "==", 0) == (False, -1) # None # Error case with invalid operator with pytest.raises(ValueError): node.compare_length([1, 2, 3], "invalid", 3)