def test_get_input_json_str():
builder = ContextBuilder('test_function_context')
builder.input_ = json.dumps({'city': 'Seattle'})
context = DurableOrchestrationContext.from_json(builder.to_json_string())
result = context.get_input()
assert 'Seattle' == result['city']
def test_get_input_string():
builder = ContextBuilder('test_function_context')
builder.input_ = json.dumps('Seattle')
context = DurableOrchestrationContext.from_json(builder.to_json_string())
assert 'Seattle' == context.get_input()
