def test_save_load(self): old_iters = params.get_parameter('iterations') params.dump_parameters('unit_test_params.pkl') params.set_parameter('iterations', 200) self.assertEqual(params.get_parameter('iterations'), 200) params.load_parameters('unit_test_params.pkl') self.assertEqual(params.get_parameter('iterations'), old_iters)
def test_get_set(self): self.assertEqual(params.get_parameter('iterations'), params._ALGO['iterations']) params.set_parameter('iterations', 200) self.assertEqual(params.get_parameter('iterations'), 200)