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)