File size: 1,468 Bytes
c508d7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from __future__ import absolute_import, division, print_function, unicode_literals
import os
import sys
def removeWhiteChar(string):
return string.replace(' ', '').replace('\t', '').replace('\n', '')
def compare_proto(str_to_compare, function_ptr):
module_id = function_ptr.__class__.__module__
functionName = function_ptr.id().split('.')[-1]
test_file = os.path.realpath(sys.modules[module_id].__file__)
expected_file = os.path.join(os.path.dirname(test_file),
"expect",
module_id.split('.')[-1] + '.' + functionName + ".expect")
print("expected_file: %s" % expected_file)
assert os.path.exists(expected_file)
with open(expected_file) as f:
expected = f.read()
str_to_compare = str(str_to_compare)
print("str_to_compare:", removeWhiteChar(str_to_compare))
print("expected:", removeWhiteChar(expected))
assert removeWhiteChar(str_to_compare) == removeWhiteChar(expected)
def write_proto(str_to_compare, function_ptr):
module_id = function_ptr.__class__.__module__
functionName = function_ptr.id().split('.')[-1]
test_file = os.path.realpath(sys.modules[module_id].__file__)
expected_file = os.path.join(os.path.dirname(test_file),
"expect",
module_id.split('.')[-1] + '.' + functionName + ".expect")
print(expected_file)
with open(expected_file, 'w') as f:
f.write(str(str_to_compare))
|