class TestParameterHelper(unittest.TestCase): def setUp(self): self.helper = ParameterHelper(args=[]) def test_check_float(self): self.assertRaises(InvalidParameter, self.helper.check_float, f=1) self.assertRaises(InvalidParameter, self.helper.check_float, f='a') self.assertEqual(1.0, self.helper.check_float(f=1.0)) self.assertEqual(1.0, self.helper.check_float(f='1')) self.assertEqual(1.0, self.helper.check_float(f='1.0')) def test_check_integer(self): self.assertRaises(InvalidParameter, self.helper.check_integer, f=1.2) self.assertRaises(InvalidParameter, self.helper.check_integer, f='1.0') self.assertEqual(1, self.helper.check_integer(f=1)) self.assertEqual(1, self.helper.check_integer(f='1')) def test_check_boolean(self): self.assertRaises(InvalidParameter, self.helper.check_boolean, f=2) self.assertRaises(InvalidParameter, self.helper.check_boolean, f='2') self.assertRaises(InvalidParameter, self.helper.check_boolean, f='yes') self.assertFalse(self.helper.check_boolean(f=False)) self.assertFalse(self.helper.check_boolean(f='false')) self.assertFalse(self.helper.check_boolean(f='0')) self.assertTrue(self.helper.check_boolean(f=True)) self.assertTrue(self.helper.check_boolean(f='true')) self.assertTrue(self.helper.check_boolean(f='1'))