def test_get_int_with_choices(self): d = { 'a': 25, 'b': 35, } b = ParseDict(d) o = [0, 25, 50, 75, 100] self.assertEqual(b.get_int('a', 50, choices=o), 25) self.assertEqual(b.get_int('b', 50, choices=o), 50)
def test_get_int(self): d = { 'a': 1, 'b': None, 'c': int(4), 'd': True, 'e': False, 'f': '3', 'g': '3.5', } b = ParseDict(d) self.assertEqual(b.get_int('a'), 1) self.assertEqual(b.get_int('b'), 0) self.assertEqual(b.get_int('b', 2), 2) self.assertEqual(b.get_int('c'), 4) self.assertEqual(b.get_int('d', 2), 1) self.assertEqual(b.get_int('e', 2), 0) self.assertEqual(b.get_int('f', 2), 3) self.assertEqual(b.get_int('g', 2), 2)