def test_extract_code_from_args_valid(self): expected_code = "some code" expected_code_bytes = str.encode(expected_code) args = {"code": [expected_code_bytes]} assert extract_code_from_args(args) == expected_code
def test_extract_code_from_args_code_is_char(self): expected_code = "a" expected_code_bytes = str.encode(expected_code) args = {"code": [expected_code_bytes]} assert extract_code_from_args(args) == expected_code