Fix failing tests

Relate-To: #112, #113

Author: michalharakal

…/ai/net/io/data/mnist/MNISTLoaderTest.kt …k/ainet/io/data/mnist/MNISTLoaderTest.ktskainet-data/skainet-data-simple/src/jvmTest/kotlin/sk/ai/net/io/data/mnist/MNISTLoaderTest.kt renamed to skainet-data/skainet-data-simple/src/jvmTest/kotlin/sk/ainet/io/data/mnist/MNISTLoaderTest.kt skainet-data/skainet-data-simple/src/jvmTest/kotlin/sk/ai/net/io/data/mnist/MNISTLoaderTest.kt renamed to skainet-data/skainet-data-simple/src/jvmTest/kotlin/sk/ainet/io/data/mnist/MNISTLoaderTest.kt

@@ -1,11 +1,18 @@
-package sk.ai.net.io.data.mnist
+package sk.ainet.io.data.mnist
 
 import kotlinx.coroutines.runBlocking
+import sk.ainet.data.mnist.MNISTConstants
+import sk.ainet.data.mnist.MNISTLoaderConfig
+import sk.ainet.data.mnist.MNISTLoaderFactory
 import kotlin.test.Test
 import kotlin.test.assertEquals
 import kotlin.test.assertNotNull
 import kotlin.test.assertTrue
 import java.io.File
+import kotlin.io.path.ExperimentalPathApi
+import kotlin.io.path.absolute
+import kotlin.io.path.createTempDirectory
+import kotlin.io.path.deleteRecursively
 
 /**
  * Tests for the MNIST loader.
@@ -22,27 +29,27 @@ class MNISTLoaderTest {
         try {
             // Create a loader with the temporary directory
             val loader = MNISTLoaderFactory.create(tempDir.absolutePath)
-            
+
             // Load the training data
             val dataset = loader.loadTrainingData()
-            
+
             // Verify the dataset
             assertNotNull(dataset)
             assertTrue(dataset.size > 0)
             assertEquals(60000, dataset.size) // MNIST training set has 60,000 images
-            
+
             // Verify the first image
             val firstImage = dataset.images[0]
             assertNotNull(firstImage)
             assertEquals(MNISTConstants.IMAGE_PIXELS, firstImage.image.size)
             assertTrue(firstImage.label >= 0 && firstImage.label <= 9)
-            
+
             // Verify that the cache files were created
             val trainImagesFile = File(tempDir, MNISTConstants.TRAIN_IMAGES_FILENAME.removeSuffix(".gz"))
             val trainLabelsFile = File(tempDir, MNISTConstants.TRAIN_LABELS_FILENAME.removeSuffix(".gz"))
             assertTrue(trainImagesFile.exists())
             assertTrue(trainLabelsFile.exists())
-            
+
             // Load the data again to test caching
             val cachedDataset = loader.loadTrainingData()
             assertEquals(dataset.size, cachedDataset.size)
@@ -55,40 +62,42 @@ class MNISTLoaderTest {
     /**
      * Tests loading the MNIST test dataset.
      */
+    @OptIn(ExperimentalPathApi::class)
     @Test
     fun testLoadTestData() = runBlocking {
         // Create a temporary directory for caching
-        val tempDir = createTempDir("mnist-test")
+        val tempDirPath = createTempDirectory("mnist-test")
+        val tempDir = tempDirPath.absolute().toString()
         try {
             // Create a loader with the temporary directory
-            val loader = MNISTLoaderFactory.create(tempDir.absolutePath)
-            
+            val loader = MNISTLoaderFactory.create(tempDir)
+
             // Load the test data
             val dataset = loader.loadTestData()
-            
+
             // Verify the dataset
             assertNotNull(dataset)
             assertTrue(dataset.size > 0)
             assertEquals(10000, dataset.size) // MNIST test set has 10,000 images
-            
+
             // Verify the first image
             val firstImage = dataset.images[0]
             assertNotNull(firstImage)
             assertEquals(MNISTConstants.IMAGE_PIXELS, firstImage.image.size)
             assertTrue(firstImage.label >= 0 && firstImage.label <= 9)
-            
+
             // Verify that the cache files were created
             val testImagesFile = File(tempDir, MNISTConstants.TEST_IMAGES_FILENAME.removeSuffix(".gz"))
             val testLabelsFile = File(tempDir, MNISTConstants.TEST_LABELS_FILENAME.removeSuffix(".gz"))
             assertTrue(testImagesFile.exists())
             assertTrue(testLabelsFile.exists())
-            
+
             // Load the data again to test caching
             val cachedDataset = loader.loadTestData()
             assertEquals(dataset.size, cachedDataset.size)
         } finally {
             // Clean up
-            tempDir.deleteRecursively()
+            tempDirPath.deleteRecursively()
         }
     }
 
@@ -99,13 +108,13 @@ class MNISTLoaderTest {
     fun testDatasetSubset() = runBlocking {
         // Create a loader with the default configuration
         val loader = MNISTLoaderFactory.create()
-        
+
         // Load the training data
         val dataset = loader.loadTrainingData()
-        
+
         // Create a subset
         val subset = dataset.subset(0, 100)
-        
+
         // Verify the subset
         assertEquals(100, subset.size)
         assertEquals(dataset.images[0], subset.images[0])
@@ -123,7 +132,7 @@ class MNISTLoaderTest {
             useCache = false
         )
         val loader = MNISTLoaderFactory.create(config)
-        
+
         // Verify that the loader was created successfully
         assertNotNull(loader)
     }