def test_udf_io(self):
udf_io = UdfIO('name', ColumnType.FLOAT, True, [2, 3], True, 1)
self.assertEqual(udf_io.id, None)
self.assertEqual(udf_io.udf_id, 1)
self.assertEqual(udf_io.is_input, True)
self.assertEqual(udf_io.is_nullable, True)
self.assertEqual(udf_io.array_dimensions, [2, 3])
self.assertEqual(udf_io.name, 'name')
self.assertEqual(udf_io.type, ColumnType.FLOAT)
def test_udf_io(self):
udf_io = UdfIO('name', ColumnType.NDARRAY, True, NdArrayType.UINT8,
[2, 3], True, 1)
self.assertEqual(udf_io.id, None)
self.assertEqual(udf_io.udf_id, 1)
self.assertEqual(udf_io.is_input, True)
self.assertEqual(udf_io.is_nullable, True)
self.assertEqual(udf_io.array_type, NdArrayType.UINT8)
self.assertEqual(udf_io.array_dimensions, [2, 3])
self.assertEqual(udf_io.name, 'name')
self.assertEqual(udf_io.type, ColumnType.NDARRAY)
def udf_io(
self, io_name: str, data_type: ColumnType,
dimensions: List[int], is_input: bool):
"""Constructs an in memory udf_io object with given info.
This function won't commit this object in the catalog database.
If you want to commit it into catalog call create_udf with
corresponding udf_id and io list
Arguments:
name(str): io name to be created
data_type(ColumnType): type of io created
dimensions(List[int]):dimensions of the io created
is_input(bool): whether a input or output, if true it is an input
"""
return UdfIO(io_name, data_type,
array_dimensions=dimensions, is_input=is_input)
def test_udf_io_equality(self):
udf_io = UdfIO('name', ColumnType.FLOAT, True, [2, 3], True, 1)
self.assertEqual(udf_io, udf_io)
udf_io2 = UdfIO('name2', ColumnType.FLOAT, True, [2, 3], True, 1)
self.assertNotEqual(udf_io, udf_io2)
udf_io2 = UdfIO('name', ColumnType.INTEGER, True, [2, 3], True, 1)
self.assertNotEqual(udf_io, udf_io2)
udf_io2 = UdfIO('name', ColumnType.FLOAT, False, [2, 3], True, 1)
self.assertNotEqual(udf_io, udf_io2)
udf_io2 = UdfIO('name', ColumnType.FLOAT, True, [2, 3, 4], True, 1)
self.assertNotEqual(udf_io, udf_io2)
udf_io2 = UdfIO('name', ColumnType.FLOAT, True, [2, 3], False, 1)
self.assertNotEqual(udf_io, udf_io2)
udf_io2 = UdfIO('name', ColumnType.FLOAT, True, [2, 3], True, 2)
self.assertNotEqual(udf_io, udf_io2)