Register test data as SampleDataSource

CMakeLists.txt

@@ -77,6 +77,7 @@ endif()
 
 #add_subdirectory(CLI/ComputeMeanShapes)
 add_subdirectory(CLI/SurfaceFeaturesExtractor)
+add_subdirectory(ShapeVariationAnalyzerSampleData)
 ## NEXT_MODULE
 
 

ShapeVariationAnalyzer/CMakeLists.txt

@@ -44,6 +44,4 @@ if(BUILD_TESTING)
   # Note that the test will also be available at runtime.
   slicer_add_python_unittest(SCRIPT ${MODULE_NAME}.py)
 
-  # Additional build-time testing
-  add_subdirectory(Testing)
 endif()

ShapeVariationAnalyzer/ShapeVariationAnalyzer.py

@@ -11,6 +11,7 @@
 import math
 import shutil
 
+import glob
 
 import pickle
 import numpy as np
@@ -2553,9 +2554,22 @@ def runTest(self):
 
     def test_ShapeVariationAnalyzer(self):
 
+
+        sampleDataLogic = slicer.modules.sampledata.widgetRepresentation().self().logic
+        sampleDataLogic.downloadSample("ShapeVariationAnalyzerSampleData")
+
         logic = ShapeVariationAnalyzerLogic()
-        input_dir = os.path.join(logic.dataPath(), "Input")
-        filepath_in = os.path.join(input_dir, "test.csv")
+
+        base_dir = slicer.mrmlScene.GetCacheManager().GetRemoteCacheDirectory() + '/ShapeVariationAnalyzerSampleData/'
+
+        # The csv file requires paths to the vtk files so let's create the csv since the path wouldn't be known beforehand
+        vtk_files = glob.glob(base_dir + '*.vtk')
+
+        filepath_in = f"{base_dir}test.csv"
+        with open(filepath_in, "w") as csv_file:
+            csv_file.write("VTK Files,Group\n")
+	    for vtk_file in vtk_files:
+	        csv_file.write(f"{vtk_file},0\n")
 
         # Test of all the groups
         keygroup = "All"

ShapeVariationAnalyzerSampleData/CMakeLists.txt

@@ -0,0 +1,27 @@
+#-----------------------------------------------------------------------------
+set(MODULE_NAME ShapeVariationAnalyzerSampleData)
+
+#-----------------------------------------------------------------------------
+set(MODULE_PYTHON_SCRIPTS
+  ${MODULE_NAME}.py
+  )
+
+set(MODULE_PYTHON_RESOURCES
+  Resources/Icons/bendicon.png
+  )
+
+#-----------------------------------------------------------------------------
+slicerMacroBuildScriptedModule(
+  NAME ${MODULE_NAME}
+  SCRIPTS ${MODULE_PYTHON_SCRIPTS}
+  RESOURCES ${MODULE_PYTHON_RESOURCES}
+  WITH_GENERIC_TESTS
+  )
+
+#-----------------------------------------------------------------------------
+if(BUILD_TESTING)
+
+  # Register the unittest subclass in the main script as a ctest.
+  # Note that the test will also be available at runtime.
+  slicer_add_python_unittest(SCRIPT ${MODULE_NAME}.py)
+endif()

ShapeVariationAnalyzerSampleData/ShapeVariationAnalyzerSampleData.py

@@ -0,0 +1,57 @@
+import os
+import unittest
+import vtk, qt, ctk, slicer
+from slicer.ScriptedLoadableModule import *
+import logging
+from pathlib import Path
+import shutil
+
+#
+# ShapeVariationAnalyzerSampleData
+#
+
+class ShapeVariationAnalyzerSampleData(ScriptedLoadableModule):
+  """Uses ScriptedLoadableModule base class, available at:
+  https://github.com/Slicer/Slicer/blob/master/Base/Python/slicer/ScriptedLoadableModule.py
+  """
+
+  def __init__(self, parent):
+    ScriptedLoadableModule.__init__(self, parent)
+    self.parent.title = "ShapeVariationAnalyzerSampleData"
+    self.parent.categories = ["TestData"]
+    self.parent.dependencies = ["SampleData"]
+    self.parent.helpText = """This module adds sample data for ShapeVariationAnalyzer"""
+
+    # don't show this module - additional data will be shown in SampleData module
+    parent.hidden = True 
+
+    import SampleData    
+
+    iconsPath = os.path.join(os.path.dirname(self.parent.path), 'Resources/Icons/')
+
+    SampleData.SampleDataLogic.registerCustomSampleDataSource(
+      sampleName='ShapeVariationAnalyzerSampleData',
+      category='ShapeVariationAnalyzer',
+      uris='https://data.kitware.com/api/v1/item/626850ed4acac99f42121c7f/download',
+      loadFiles=True, # Unzip it
+      fileNames='ShapeVariationAnalyzerSampleData.zip',
+      loadFileType='ZipFile',
+      nodeNames='ShapeVariationAnalyzerSampleData',
+      checksums='SHA512:c2a8b8b33b18bb9ad6308afd2c557a7743edaf77f3e03d50f4df0059d6aa4cdd78093dda144ed2f0a27f088464e30d6e654fc68cd19b37b6027e44bebf97bfc3',
+      thumbnailFileName=os.path.join(iconsPath, 'bendicon.png'),
+      customDownloader=self.downloadSampleDataInFolder,
+    )
+
+  @staticmethod
+  def downloadSampleDataInFolder(source):
+
+    outPath = slicer.mrmlScene.GetCacheManager().GetRemoteCacheDirectory()
+
+    if not os.path.exists(outPath):
+      os.mkdir(outPath)
+
+    sampleDataLogic = slicer.modules.sampledata.widgetRepresentation().self().logic
+
+    for uri, fileName, checksums in zip(source.uris, source.fileNames, source.checksums):
+      sampleDataLogic.downloadFile(uri, outPath, fileName)
+