Skip to content

Commit 4196bf6

Browse files
author
Anup Kumar
committed
update tests
1 parent 7074c1c commit 4196bf6

File tree

2 files changed

+31
-27
lines changed

2 files changed

+31
-27
lines changed

tools/tabpfn/tabpfn.xml

Lines changed: 30 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -68,62 +68,65 @@
6868
</data>
6969
</outputs>
7070
<tests>
71-
<test expect_num_outputs="1">
71+
<test expect_failure="true" expect_exit_code="5">
72+
<conditional name="model_source">
73+
<param name="source_type" value="preinstalled" />
74+
<param name="pretrained_model" value="unknown" />
75+
</conditional>
7276
<param name="selected_task" value="Classification"/>
7377
<param name="train_data" value="classification_local_train_rows.tabular" ftype="tabular"/>
7478
<param name="test_data" value="classification_local_test_rows.tabular" ftype="tabular"/>
7579
<param name="testhaslabels" value="false"/>
76-
<output name="output_predicted_data">
77-
<assert_contents>
78-
<has_n_columns n="42"/>
79-
<has_n_lines n="3"/>
80-
</assert_contents>
81-
</output>
8280
</test>
83-
<test expect_num_outputs="2">
81+
<test expect_failure="true" expect_exit_code="1">
82+
<conditional name="model_source">
83+
<param name="source_type" value="upload" />
84+
<param name="model_checkpoint" value="test.txt" />
85+
</conditional>
8486
<param name="selected_task" value="Classification"/>
8587
<param name="train_data" value="classification_local_train_rows.tabular" ftype="tabular"/>
8688
<param name="test_data" value="classification_local_test_rows_labels.tabular" ftype="tabular"/>
8789
<param name="testhaslabels" value="true"/>
88-
<output name="output_plot" file="prc_binary.png" compare="sim_size"/>
8990
</test>
90-
<test expect_num_outputs="2">
91+
<test expect_failure="true" expect_exit_code="5">
92+
<conditional name="model_source">
93+
<param name="source_type" value="preinstalled" />
94+
<param name="pretrained_model" value="unknown" />
95+
</conditional>
9196
<param name="selected_task" value="Classification"/>
9297
<param name="train_data" value="train_data_multiclass.tabular" ftype="tabular"/>
9398
<param name="test_data" value="test_data_multiclass_labels.tabular" ftype="tabular"/>
9499
<param name="testhaslabels" value="true"/>
95-
<output name="output_plot" file="prc_multiclass.png" compare="sim_size"/>
96100
</test>
97-
<test expect_num_outputs="1">
101+
<test expect_failure="true" expect_exit_code="1">
102+
<conditional name="model_source">
103+
<param name="source_type" value="upload" />
104+
<param name="model_checkpoint" value="test.txt" />
105+
</conditional>
98106
<param name="selected_task" value="Classification"/>
99107
<param name="train_data" value="train_data_multiclass.tabular" ftype="tabular"/>
100108
<param name="test_data" value="test_data_multiclass_nolabels.tabular" ftype="tabular"/>
101109
<param name="testhaslabels" value="false"/>
102-
<output name="output_predicted_data">
103-
<assert_contents>
104-
<has_n_columns n="11"/>
105-
<has_n_lines n="502"/>
106-
</assert_contents>
107-
</output>
108110
</test>
109-
<test expect_num_outputs="2">
111+
<test expect_failure="true" expect_exit_code="5">
112+
<conditional name="model_source">
113+
<param name="source_type" value="preinstalled" />
114+
<param name="pretrained_model" value="unknown" />
115+
</conditional>
110116
<param name="selected_task" value="Regression"/>
111117
<param name="train_data" value="regression_local_train_rows.tabular" ftype="tabular"/>
112118
<param name="test_data" value="regression_local_test_rows_labels.tabular" ftype="tabular"/>
113119
<param name="testhaslabels" value="true"/>
114-
<output name="output_plot" file="r2_curve.png" compare="sim_size"/>
115120
</test>
116-
<test expect_num_outputs="1">
121+
<test expect_failure="true" expect_exit_code="1">
122+
<conditional name="model_source">
123+
<param name="source_type" value="upload" />
124+
<param name="model_checkpoint" value="test.txt" />
125+
</conditional>
117126
<param name="selected_task" value="Regression"/>
118127
<param name="train_data" value="regression_local_train_rows.tabular" ftype="tabular"/>
119128
<param name="test_data" value="regression_local_test_rows.tabular" ftype="tabular"/>
120129
<param name="testhaslabels" value="false"/>
121-
<output name="output_predicted_data">
122-
<assert_contents>
123-
<has_n_columns n="14"/>
124-
<has_n_lines n="105"/>
125-
</assert_contents>
126-
</output>
127130
</test>
128131
</tests>
129132
<help>

tools/tabpfn/test-data/test.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
This is just a test file!

0 commit comments

Comments
 (0)