diff options
Diffstat (limited to 'test/sql/test_functions.py')
-rw-r--r-- | test/sql/test_functions.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/test/sql/test_functions.py b/test/sql/test_functions.py index fc227f375..f0fcd4b72 100644 --- a/test/sql/test_functions.py +++ b/test/sql/test_functions.py @@ -127,6 +127,35 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): "my_func(:param_1, :param_2, :param_3)" ) + def test_custom_registered_identifier(self): + class GeoBuffer(GenericFunction): + type = Integer + package = "geo" + name = "BufferOne" + identifier = "buf1" + + class GeoBuffer2(GenericFunction): + type = Integer + name = "BufferTwo" + identifier = "buf2" + + class BufferThree(GenericFunction): + type = Integer + identifier = "buf3" + + self.assert_compile( + func.geo.buf1(), + "BufferOne()" + ) + self.assert_compile( + func.buf2(), + "BufferTwo()" + ) + self.assert_compile( + func.buf3(), + "BufferThree()" + ) + def test_custom_args(self): class myfunc(GenericFunction): pass |