Updated test resources

jpmml · Sep 21, 2023 · a05cf9b · a05cf9b
1 parent aba4190
commit a05cf9b
Show file tree

Hide file tree

Showing 8 changed files with 19,553 additions and 2 deletions.
diff --git a/pmml-sklearn/src/test/java/org/jpmml/sklearn/testing/SkLearn2PMMLTest.java b/pmml-sklearn/src/test/java/org/jpmml/sklearn/testing/SkLearn2PMMLTest.java
@@ -129,4 +129,14 @@ public void evaluateExpressionVersicolor() throws Exception {
 	public void evaluateGBDTLRVersicolor() throws Exception {
 		evaluate("GBDTLR", VERSICOLOR);
 	}
+
+	@Test
+	public void evaluateEstimatorChainWine() throws Exception {
+		evaluate("EstimatorChain", "Wine");
+	}
+
+	@Test
+	public void evaluateSelectFirstWine() throws Exception {
+		evaluate("SelectFirst", "Wine");
+	}
 }
diff --git a/pmml-sklearn/src/test/resources/common.py b/pmml-sklearn/src/test/resources/common.py
@@ -92,3 +92,8 @@ def load_wheat(name):
 	df = load_csv(name)
 	print(df.dtypes)
 	return df
+
+def load_wine(name):
+	df = load_csv(name)
+	print(df.dtypes)
+	return df