def case_of_default(sub_case): compiler = NonCallableMock(CaseCompiler) compiler.compile_fixed.return_value = sentinel.case compiler.compile_default = Mock(return_value=sub_case) return compiler
def case(case_of_default): compiler = NonCallableMock(CaseCompiler) compiler.compile_default = Mock(return_value=case_of_default) return compiler