diff --git a/src/sql_mock/column_mocks.py b/src/sql_mock/column_mocks.py index f73311e..058595a 100644 --- a/src/sql_mock/column_mocks.py +++ b/src/sql_mock/column_mocks.py @@ -1,4 +1,4 @@ -from sql_mock.constants import NO_INPUT +from sql_mock.constants import NO_INPUT, NoInput class ColumnMock: @@ -29,7 +29,7 @@ def __init__(self, default=None, nullable=False) -> None: def to_sql(self, column_name: str, value=NO_INPUT) -> str: # Note: We compare against NO_INPUT instead of checking for None since None could be a valid input for nullable columns - val = value if value != NO_INPUT else self.default + val = value if not isinstance(value, NoInput) else self.default # In case the val is None, we convert it to NULL if val is None: return f"cast(NULL AS {self.dtype}) AS {column_name}" diff --git a/src/sql_mock/constants.py b/src/sql_mock/constants.py index 47d53c6..1d233c5 100644 --- a/src/sql_mock/constants.py +++ b/src/sql_mock/constants.py @@ -1 +1,5 @@ -NO_INPUT = "NO INPUT PROVIDED" +class NoInput: + pass + + +NO_INPUT = NoInput()