From 4d9c8ec1ecdddb4b44594efd1e2926478695717c Mon Sep 17 00:00:00 2001 From: Marcus Edel Date: Wed, 5 Apr 2017 22:10:19 +0200 Subject: [PATCH] Test on another dataset, to stabilize the ann mrpt test case, which sometimes fails on small datasets. --- tests/benchmark_ann.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/benchmark_ann.py b/tests/benchmark_ann.py index 4192f26..d200043 100644 --- 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__':