Test on another dataset, to stabilize the ann mrpt test case, which s…

…ometimes fails on small datasets.
mlpack · Apr 5, 2017 · 4d9c8ec · 4d9c8ec
1 parent ca2094a
commit 4d9c8ec
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/benchmark_ann.py b/tests/benchmark_ann.py
@@ -55,8 +55,7 @@ class ANN_MRPT_TEST(unittest.TestCase):
   Test initialization.
   '''
   def setUp(self):
-    self.dataset = ['datasets/iris_train.csv', 'datasets/iris_test.csv',
-        'datasets/iris_labels.csv']
+    self.dataset = ['datasets/optdigits_train.csv', 'datasets/optdigits_test.csv']
     self.verbose = False
     self.timeout = 9000
 
@@ -75,7 +74,7 @@ def test_Constructor(self):
   Test the 'RunMetrics' function.
   '''
   def test_RunMetrics(self):
-    result = self.instance.RunMetrics("-k 10 -n 10")
+    result = self.instance.RunMetrics("-k 5 -n 5")
     self.assertTrue(result["Runtime"] > 0)
 
 if __name__ == '__main__':