Added new plotter test

mghosh00 · Apr 14, 2024 · a6a8059 · a6a8059
1 parent f3c358d
commit a6a8059
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/javaNeuralNetwork/src/test/java/neural_network/util/PlotterTest.java b/javaNeuralNetwork/src/test/java/neural_network/util/PlotterTest.java
@@ -242,4 +242,14 @@ void plotLossNoMocks() throws IOException {
         assertNull(Plotter.getChart());
         assertNull(Plotter.getWrappedChart());
     }
+
+    @Test
+    void plotLossNewDir() throws IOException {
+        Plotter.setDirName(dirName + "losses/");
+        Plotter.plotLoss(lossDf, "");
+        assertNull(Plotter.getChart());
+        assertNull(Plotter.getWrappedChart());
+        Files.deleteIfExists(Path.of(dirName + "losses/losses.png"));
+        Files.deleteIfExists(Path.of(dirName + "losses"));
+    }
 }
diff --git a/javaNeuralNetwork/src/test/resources/util/plots/losses.png b/javaNeuralNetwork/src/test/resources/util/plots/losses.png
diff --git a/javaNeuralNetwork/src/test/resources/util/plots/true/comparison.png b/javaNeuralNetwork/src/test/resources/util/plots/true/comparison.png