def test_can_register_class_as_param_source(self):
source_name = "params-test-class-param-source"
params.register_param_source_for_name(source_name, ParamsRegistrationTests.ParamSourceClass)
source = params.param_source_for_name(source_name, None, {"parameter": 42})
self.assertEqual({"class-key": 42}, source.params())
params._unregister_param_source_for_name(source_name)
def test_can_register_class_as_param_source(self):
source_name = "params-test-class-param-source"
params.register_param_source_for_name(source_name, ParamsRegistrationTests.ParamSourceClass)
source = params.param_source_for_name(source_name, None, {"parameter": 42})
self.assertEqual({"class-key": 42}, source.params())
params._unregister_param_source_for_name(source_name)
def test_can_register_function_as_param_source(self):
source_name = "params-test-function-param-source"
params.register_param_source_for_name(source_name, ParamsRegistrationTests.param_source_function)
source = params.param_source_for_name(source_name, None, {"parameter": 42})
self.assertEqual({"key": 42}, source.params())
params._unregister_param_source_for_name(source_name)
def test_can_register_function_as_param_source(self):
source_name = "params-test-function-param-source"
params.register_param_source_for_name(source_name, ParamsRegistrationTests.param_source_function)
source = params.param_source_for_name(source_name, None, {"parameter": 42})
self.assertEqual({"key": 42}, source.params())
params._unregister_param_source_for_name(source_name)
