1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
import six
def equal_codes(code, ref_code):
assert isinstance(code, list) or isinstance(code, six.string_types)
assert isinstance(ref_code, list) or isinstance(code, six.string_types)
if isinstance(code, six.string_types):
code = code.splitlines()
if isinstance(ref_code, six.string_types):
ref_code = ref_code.splitlines()
code = [line.strip() for line in code if len(line)]
ref_code = [line.strip() for line in ref_code if len(line)]
if six.PY3:
ref_code = [line.replace("\xC2\xB7", "\u00B7") for line in ref_code]
return code == ref_code
|
