@@ -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