tmp

Author: e-strauss

src/test/java/org/apache/sysds/test/functions/jmlc/JMLConnectionTest.java

@@ -30,6 +30,7 @@
 import org.apache.sysds.runtime.frame.data.FrameBlock;
 import org.apache.sysds.runtime.io.IOUtilFunctions;
 import org.apache.sysds.runtime.matrix.data.MatrixBlock;
+import org.apache.sysds.runtime.util.HDFSTool;
 import org.apache.sysds.test.AutomatedTestBase;
 import org.junit.Assert;
 import org.junit.Test;
@@ -386,4 +387,44 @@ public void testConvertToFrame() {
 			throw new AssertionError(e);
 		}
 	}
+
+	@Test
+	public void testTransformMetaData1() {
+		try (Connection conn = new Connection()) {
+			FrameBlock frame = conn.readTransformMetaDataFromFile(SCRIPT_DIR + TEST_DIR + "tfmtd_example2/");
+			Assert.assertEquals("Hello", frame.get(0,0));
+		} catch (IOException e) {
+			Assert.assertEquals("Missing transform specification.", e.getMessage());
+		}
+	}
+
+	@Test
+	public void testTransformMetaData2() {
+		try (Connection conn = new Connection()) {
+			String spec = HDFSTool.readStringFromHDFSFile(SCRIPT_DIR + TEST_DIR+"tfmtd_example2/spec.json");
+			FrameBlock frame = conn.readTransformMetaDataFromFile(spec, SCRIPT_DIR + TEST_DIR + "tfmtd_example2/");
+			Assert.assertEquals("91312·1", frame.get(0,0));
+		} catch (IOException e) {
+			throw new AssertionError(e);
+		}
+	}
+
+	@Test
+	public void testTransformMetaData3() {
+		try (Connection conn = new Connection()) {
+			conn.readTransformMetaDataFromPath(SCRIPT_DIR + TEST_DIR + "tfmtd_example2");
+		} catch (Exception e) {
+			Assert.assertEquals("NullPointerException", e.getClass().getSimpleName());
+		}
+	}
+
+	@Test
+	public void testTransformMetaData4() {
+		try (Connection conn = new Connection()) {
+			String spec = HDFSTool.readStringFromHDFSFile(SCRIPT_DIR + TEST_DIR+"tfmtd_example2/spec.json");
+			conn.readTransformMetaDataFromPath(spec,SCRIPT_DIR + TEST_DIR + "tfmtd_example2");
+		} catch (Exception e) {
+			Assert.assertEquals("NullPointerException", e.getClass().getSimpleName());
+		}
+	}
 }