itk_module_test()
set(ITKClassifiersTests
    itkScalarImageKmeansImageFilterTest.cxx
    itkSupervisedImageClassifierTest.cxx
    itkBayesianClassifierImageFilterTest.cxx
    itkKmeansModelEstimatorTest.cxx
    itkImageClassifierFilterTest.cxx
    itkSampleClassifierFilterTest1.cxx
    itkSampleClassifierFilterTest2.cxx
    itkSampleClassifierFilterTest3.cxx
    itkSampleClassifierFilterTest4.cxx
    itkSampleClassifierFilterTest5.cxx
    itkSampleClassifierFilterTest6.cxx
    itkSampleClassifierFilterTest7.cxx
    itkScalarImageKmeansImageFilterTest.cxx
    itkScalarImageKmeansImageFilter3DTest.cxx)

createtestdriver(ITKClassifiers "${ITKClassifiers-Test_LIBRARIES}" "${ITKClassifiersTests}")

itk_add_test(
  NAME
  itkBayesianClassifierImageFilterTest1
  COMMAND
  ITKClassifiersTestDriver
  --compare
  DATA{Baseline/itkBayesianClassifierImageFilterTestLabelMapNoPriors.png}
  ${ITK_TEST_OUTPUT_DIR}/itkBayesianClassifierImageFilterTestLabelMapNoPriors.png
  itkBayesianClassifierImageFilterTest
  ${ITK_EXAMPLE_DATA_ROOT}/BrainProtonDensitySlice.png
  ${ITK_TEST_OUTPUT_DIR}/itkBayesianClassifierImageFilterTestLabelMapNoPriors.png
  4
  2
  0)
itk_add_test(
  NAME
  itkBayesianClassifierImageFilterTest2
  COMMAND
  ITKClassifiersTestDriver
  --compare
  DATA{Baseline/itkBayesianClassifierImageFilterTestLabelMapPriors.png}
  ${ITK_TEST_OUTPUT_DIR}/itkBayesianClassifierImageFilterTestLabelMapPriors.png
  itkBayesianClassifierImageFilterTest
  ${ITK_EXAMPLE_DATA_ROOT}/BrainProtonDensitySlice.png
  ${ITK_TEST_OUTPUT_DIR}/itkBayesianClassifierImageFilterTestLabelMapPriors.png
  4
  2
  1)
itk_add_test(
  NAME
  itkKmeansModelEstimatorTest
  COMMAND
  ITKClassifiersTestDriver
  itkKmeansModelEstimatorTest)
itk_add_test(
  NAME
  itkImageClassifierFilterTest
  COMMAND
  ITKClassifiersTestDriver
  --compare
  DATA{${ITK_DATA_ROOT}/Baseline/Statistics/ImageClassifierFilterTestClassifiedImageTest.png}
  ${ITK_TEST_OUTPUT_DIR}/ImageClassifierFilterTestClassifiedImage.png
  itkImageClassifierFilterTest
  ${ITK_TEST_OUTPUT_DIR}/ImageClassifierFilterTestClassifiedImage.png)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest1
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest1)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest2
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest2)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest3
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest3)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest4
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest4)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest5
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest5)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest6
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest6)
itk_add_test(
  NAME
  itkSampleClassifierFilterTest7
  COMMAND
  ITKClassifiersTestDriver
  itkSampleClassifierFilterTest7
  DATA{${ITK_DATA_ROOT}/Input/Statistics/TwoDimensionTwoGaussian.dat}
  DATA{${ITK_DATA_ROOT}/Input/Statistics/TwoDimensionTwoGaussianTesting.dat})
itk_add_test(
  NAME
  itkSupervisedImageClassifierTest
  COMMAND
  ITKClassifiersTestDriver
  itkSupervisedImageClassifierTest)
itk_add_test(
  NAME
  itkScalarImageKmeansImageFilterTest
  COMMAND
  ITKClassifiersTestDriver
  --compare
  DATA{Baseline/itkScalarImageKmeansImageFilterTest.png}
  ${ITK_TEST_OUTPUT_DIR}/itkScalarImageKmeansImageFilterTest.png
  itkScalarImageKmeansImageFilterTest
  ${ITK_EXAMPLE_DATA_ROOT}/BrainProtonDensitySlice.png
  ${ITK_TEST_OUTPUT_DIR}/itkScalarImageKmeansImageFilterTest.png
  0
  5
  10
  80
  165
  190
  220)
itk_add_test(
  NAME
  itkScalarImageKmeansImageFilter3DTest
  COMMAND
  ITKClassifiersTestDriver
  --compare
  ${ITK_EXAMPLE_DATA_ROOT}/KmeansTest_T1KmeansPrelimSegmentation.nii.gz
  ${ITK_TEST_OUTPUT_DIR}/KmeansTest_T1KmeansPrelimSegmentation.nii.gz
  itkScalarImageKmeansImageFilter3DTest
  ${ITK_EXAMPLE_DATA_ROOT}/KmeansTest_T1UCharRaw.nii.gz
  ${ITK_EXAMPLE_DATA_ROOT}/KmeansTest_T1RawSkullStrip.nii.gz
  ${ITK_TEST_OUTPUT_DIR}/KmeansTest_T1KmeansPrelimSegmentation.nii.gz)