diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 12:24:33 -0500 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 12:24:33 -0500 |
| commit | 61423e7d4b98eeb8bb73ebb5786bd1477d99ee23 (patch) | |
| tree | 0ab7f1fa1c05b7451d7b7cd3cd867b93f0b988f2 /test | |
| parent | Extract durations array (diff) | |
| download | pygensvm-61423e7d4b98eeb8bb73ebb5786bd1477d99ee23.tar.gz pygensvm-61423e7d4b98eeb8bb73ebb5786bd1477d99ee23.zip | |
Add support for interrupted grid search
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_util.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/test/test_util.py b/test/test_util.py index 7a6f5bb..b65a694 100644 --- a/test/test_util.py +++ b/test/test_util.py @@ -8,6 +8,7 @@ from __future__ import division, print_function +import numpy as np import unittest from gensvm.util import get_ranks @@ -27,3 +28,14 @@ class GenSVMUtilTestCase(unittest.TestCase): x = [-1, -2, -3] self.assertEqual(get_ranks(x), [3, 2, 1]) + + def test_get_ranks_nan(self): + """ UTIL: Test ranking function with NaN entries """ + x = [3, 2, 1, 4, 5, np.nan] + self.assertEqual(get_ranks(x), [3, 2, 1, 4, 5, 6]) + + x = [3, 2, 1, np.nan, 4, 5] + self.assertEqual(get_ranks(x), [3, 2, 1, 6, 4, 5]) + + x = [3, 2, 1, np.nan, 4, 5, np.nan] + self.assertEqual(get_ranks(x), [3, 2, 1, 6, 4, 5, 6]) |