add test for prc curve (#1566)

bgruening · Jan 15, 2025 · 0544107 · 0544107
1 parent bce8b02
commit 0544107
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/tools/tabpfn/tabpfn.xml b/tools/tabpfn/tabpfn.xml
@@ -28,7 +28,7 @@
     </inputs>
     <outputs>
         <data format="tabular" name="output_predicted_data" from_work_dir="output_predicted_data" label="Predicted data"></data>
-        <data format="png" name="output_prec_recall_curve" from_work_dir="output_prec_recall_curve" label="Precision-recall curve"></data>
+        <data format="png" name="output_prec_recall_curve" from_work_dir="output_prec_recall_curve.png" label="Precision-recall curve"></data>
     </outputs>
     <tests>
         <test>
@@ -41,6 +41,11 @@
 	        </assert_contents>
 	    </output>
         </test>
+	<test>
+	    <param name="train_data" value="local_train_rows.tabular" ftype="tabular" />
+	    <param name="test_data" value="local_test_rows.tabular" ftype="tabular" />
+	    <output name="output_prec_recall_curve" file="pr_curve.png" compare="sim_size" />
+	</test>
     </tests>
     <help>
         <![CDATA[

diff --git a/tools/tabpfn/test-data/pr_curve.png b/tools/tabpfn/test-data/pr_curve.png