aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/training/python/training/hparam_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/training/python/training/hparam_test.py')
-rw-r--r--tensorflow/contrib/training/python/training/hparam_test.py31
1 files changed, 1 insertions, 30 deletions
diff --git a/tensorflow/contrib/training/python/training/hparam_test.py b/tensorflow/contrib/training/python/training/hparam_test.py
index 949c262f5b..f54514cefd 100644
--- a/tensorflow/contrib/training/python/training/hparam_test.py
+++ b/tensorflow/contrib/training/python/training/hparam_test.py
@@ -318,42 +318,13 @@ class HParamsTest(test.TestCase):
self.assertEqual(3.0, hparams.b)
self.assertEqual('relu4', hparams.c_c)
- def testSetHParamListNonListMismatch(self):
+ def testSetHParamTypeMismatch(self):
hparams = hparam.HParams(a=1, b=[2.0, 3.0])
with self.assertRaisesRegexp(ValueError, r'Must not pass a list'):
hparams.set_hparam('a', [1.0])
with self.assertRaisesRegexp(ValueError, r'Must pass a list'):
hparams.set_hparam('b', 1.0)
- def testSetHParamTypeMismatch(self):
- hparams = hparam.HParams(
- int_=1, str_='str', bool_=True, float_=1.1, list_int=[1, 2], none=None)
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('str_', 2.2)
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('int_', False)
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('bool_', 1)
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('int_', 2.2)
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('list_int', [2, 3.3])
-
- with self.assertRaises(ValueError):
- hparams.set_hparam('int_', '2')
-
- # Casting int to float is OK
- hparams.set_hparam('float_', 1)
-
- # Getting stuck with NoneType :(
- hparams.set_hparam('none', '1')
- self.assertEqual('1', hparams.none)
-
def testNonProtoFails(self):
with self.assertRaisesRegexp(AssertionError, ''):
hparam.HParams(hparam_def=1)