Skip to content

Commit 008ba58

Browse files
Corrected variable names (#28)
1 parent 4180c95 commit 008ba58

File tree

2 files changed

+31
-2
lines changed

2 files changed

+31
-2
lines changed

exasol_udf_mock_python/mock_meta_data.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def input_type(self):
154154
return self._input_type
155155

156156
@property
157-
def input_columns_count(self):
157+
def input_column_count(self):
158158
return self._input_column_count
159159

160160
@property
@@ -166,7 +166,7 @@ def output_type(self):
166166
return self._output_type
167167

168168
@property
169-
def output_columns_count(self):
169+
def output_column_count(self):
170170
return self._output_column_count
171171

172172
@property

tests/test_executor_exa_meta.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,3 +62,32 @@ def run(ctx):
6262
exa = MockExaEnvironment(meta, connections={"TEST_CON": Connection(address="https://test.de")})
6363
result = executor.run([Group([(1,)])], exa)
6464
assert result == [Group([("https://test.de",)])]
65+
66+
67+
def test_exa_meta_column_count():
68+
def udf_wrapper():
69+
script_code = exa.meta.script_code
70+
71+
def run(ctx):
72+
ctx.emit(script_code)
73+
74+
input_columns = [Column("t1", int, "INTEGER"),
75+
Column("t2", int, "INTEGER"),
76+
Column("t3", int, "INTEGER")]
77+
output_columns = [Column("o1", str, "VARCHAR(2000)"),
78+
Column("o2", str, "VARCHAR(2000)")]
79+
meta = MockMetaData(
80+
script_code_wrapper_function=udf_wrapper,
81+
input_type="SET",
82+
input_columns=input_columns,
83+
output_type="EMITS",
84+
output_columns=output_columns
85+
)
86+
exa = MockExaEnvironment(meta)
87+
88+
assert exa.meta.input_column_count == len(input_columns)
89+
assert len(exa.meta.input_columns) == len(input_columns)
90+
assert exa.meta.input_columns == input_columns
91+
assert exa.meta.output_column_count == len(output_columns)
92+
assert len(exa.meta.output_columns) == len(output_columns)
93+
assert exa.meta.output_columns == output_columns

0 commit comments

Comments
 (0)