def test_NumberRange(self):
"""
Test for class IntegerRange and FloatRange.
"""
dtypes = {int: np.int32, float: np.float32}
expected_linspace = [1, 2, 3, 4, 5, 6, 7, 8, 9]
number_range_linspace = IntegerRange(start=1, stop=9, num=9, range_type="linspace")
number_range_linspace.transform()
self.assertListEqual(expected_linspace, number_range_linspace.values)
expected_geomspace = [1, 10, 100, 1000, 10000]
number_range_geomspace = IntegerRange(1, 10000, num=5, range_type="geomspace")
number_range_geomspace.transform()
self.assertListEqual(expected_geomspace, number_range_geomspace.values)
number_range_range = IntegerRange(self.start, self.end, step=2, range_type="range")
number_range_range.transform()
self.assertListEqual(number_range_range.values, list(np.arange(self.start, self.end, 2)))
number_range_logspace = FloatRange(-1, 1, num=50, range_type='logspace')
number_range_logspace.transform()
np.testing.assert_array_almost_equal(number_range_logspace.values, np.logspace(-1, 1, num=50).tolist())
# error tests
with self.assertRaises(ValueError):
number_range = IntegerRange(start=0, stop=self.end, range_type="geomspace")
number_range.transform()
with self.assertRaises(ValueError):
number_range = IntegerRange(start=1, stop=15, range_type="logspace")
number_range.transform()
with self.assertRaises(ValueError):
IntegerRange(start=self.start, stop=self.end, range_type="ownspace")
class BaseTest(unittest.TestCase):
def setUp(self):
"""
Set default start setting for all tests.
"""
self.intger_range = IntegerRange(2, 6)
self.float_range = FloatRange(0.1, 5.7)
self.categorical = Categorical(
["a", "b", "c", "d", "e", "f", "g", "h"])
self.bool = BooleanSwitch()
def test_rand_success(self):
for _ in range(100):
self.assertIn(self.intger_range.get_random_value(),
list(range(2, 6)))
self.assertGreaterEqual(self.float_range.get_random_value(), 0.1)
self.assertLess(self.float_range.get_random_value(), 5.7)
self.assertIn(
self.categorical.get_random_value(),
["a", "b", "c", "d", "e", "f", "g", "h"],
)
self.assertIn(self.bool.get_random_value(), [True, False])
self.float_range.transform()
self.intger_range.transform()
for _ in range(100):
self.assertIn(
self.intger_range.get_random_value(definite_list=True),
self.intger_range.values,
)
self.assertIn(
self.float_range.get_random_value(definite_list=True),
self.float_range.values,
)
def test_rand_error(self):
with self.assertRaises(ValueError):
self.intger_range.get_random_value(definite_list=True)
self.float_range.get_random_value(definite_list=True)
self.bool.get_random_value(definite_list=True)
self.categorical.get_random_value(definite_list=True)
class HyperparameterBaseTest(unittest.TestCase):
def setUp(self):
"""
Set default start setting for all tests.
"""
self.intger_range = IntegerRange(2, 6)
self.float_range = FloatRange(0.1, 5.7)
self.cateogrical_truth = ["a", "b", "c", "d", "e", "f", "g", "h"]
self.categorical = Categorical(self.cateogrical_truth)
self.bool = BooleanSwitch()
def test_rand_success(self):
for _ in range(100):
self.assertIn(self.intger_range.get_random_value(),
list(range(2, 6)))
self.assertGreaterEqual(self.float_range.get_random_value(), 0.1)
self.assertLess(self.float_range.get_random_value(), 5.7)
self.assertIn(self.categorical.get_random_value(),
self.cateogrical_truth)
self.assertIn(self.bool.get_random_value(), [True, False])
self.float_range.transform()
self.intger_range.transform()
for _ in range(100):
self.assertIn(
self.intger_range.get_random_value(definite_list=True),
self.intger_range.values)
self.assertIn(
self.float_range.get_random_value(definite_list=True),
self.float_range.values)
def test_domain(self):
self.float_range.transform()
self.intger_range.transform()
self.assertListEqual(self.intger_range.values, list(np.arange(2, 6)))
self.assertListEqual(self.float_range.values,
list(np.linspace(0.1, 5.7, dtype=np.float64)))
big_float_range = FloatRange(-300.57, np.pi * 4000)
big_float_range.transform()
self.assertListEqual(big_float_range.values,
list(np.linspace(-300.57, np.pi * 4000)))
self.assertListEqual(self.categorical.values,
["a", "b", "c", "d", "e", "f", "g", "h"])
self.assertListEqual(self.bool.values, [True, False])
def test_rand_error(self):
with self.assertRaises(ValueError):
self.intger_range.get_random_value(definite_list=True)
with self.assertRaises(ValueError):
self.float_range.get_random_value(definite_list=True)
with self.assertRaises(NotImplementedError):
self.categorical.get_random_value(definite_list=False)
with self.assertRaises(NotImplementedError):
self.categorical.get_random_value(definite_list=False)
def test_categorical(self):
self.assertEqual(self.categorical[2], self.cateogrical_truth[2])
