def test_create_udf_io_object(self, udfio_mock):
catalog = CatalogManager()
actual = catalog.udf_io('name', ColumnType.TEXT, [100], True)
udfio_mock.assert_called_with('name',
ColumnType.TEXT,
array_dimensions=[100],
is_input=True)
self.assertEqual(actual, udfio_mock.return_value)
def test_create_udf_io_object(self, udfio_mock):
catalog = CatalogManager()
actual = catalog.udf_io('name', ColumnType.NDARRAY, NdArrayType.UINT8,
[2, 3, 4], True)
udfio_mock.assert_called_with('name',
ColumnType.NDARRAY,
array_type=NdArrayType.UINT8,
array_dimensions=[2, 3, 4],
is_input=True)
self.assertEqual(actual, udfio_mock.return_value)
