diff options
Diffstat (limited to 'test/sql/test_functions.py')
| -rw-r--r-- | test/sql/test_functions.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/test/sql/test_functions.py b/test/sql/test_functions.py index 5769e4a1a..b69f8f6ba 100644 --- a/test/sql/test_functions.py +++ b/test/sql/test_functions.py @@ -38,7 +38,7 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): __return_type__ = sqltypes.Integer def __init__(self, arg, **kwargs): - GenericFunction.__init__(self, args=[arg], **kwargs) + GenericFunction.__init__(self, arg, **kwargs) self.assert_compile( fake_func('foo'), @@ -114,6 +114,15 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): assert isinstance(func.mypackage.myfunc(), f1) assert isinstance(func.myotherpackage.myfunc(), f2) + def test_custom_args(self): + class myfunc(GenericFunction): + pass + + self.assert_compile( + myfunc(1, 2, 3), + "myfunc(:param_1, :param_2, :param_3)" + ) + def test_namespacing_conflicts(self): self.assert_compile(func.text('foo'), 'text(:text_1)') |