Skip to content

Commit 4391bf4

Browse files
committed
Add txt test, improve tests and reformat
1 parent f131ba6 commit 4391bf4

File tree

1 file changed

+20
-7
lines changed

1 file changed

+20
-7
lines changed

tests/test.py

Lines changed: 20 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,14 @@ def check_decode(raw: bytes, name: str):
4141
svg_output_folder = os.path.join(output_folder, 'svg')
4242
json_output_folder = os.path.join(output_folder, 'json')
4343
md_output_folder = os.path.join(output_folder, 'md')
44+
txt_output_folder = os.path.join(output_folder, 'txt')
4445
html_output_folder = os.path.join(output_folder, 'html')
4546
files_folder = os.path.join(script_folder, 'files')
4647

4748
os.makedirs(svg_output_folder, exist_ok=True)
4849
os.makedirs(json_output_folder, exist_ok=True)
4950
os.makedirs(md_output_folder, exist_ok=True)
51+
os.makedirs(txt_output_folder, exist_ok=True)
5052
os.makedirs(html_output_folder, exist_ok=True)
5153

5254
if os.name == 'nt':
@@ -76,6 +78,7 @@ def check_decode(raw: bytes, name: str):
7678
svg_output_path = os.path.join(svg_output_folder, file.replace('.rm', '.svg'))
7779
json_output_path = os.path.join(json_output_folder, file.replace('.rm', '.json'))
7880
md_output_path = os.path.join(md_output_folder, file.replace('.rm', '.md'))
81+
txt_output_path = os.path.join(txt_output_folder, file.replace('.rm', '.txt'))
7982
html_output_path = os.path.join(html_output_folder, file.replace('.rm', '.html'))
8083

8184
print("Processing file:", file)
@@ -88,13 +91,13 @@ def check_decode(raw: bytes, name: str):
8891

8992
begin = time.time()
9093
success = lib.convertToJsonFile(tree_id, json_output_path.encode())
94+
if not success:
95+
exit(-1)
9196
print(f"JSON (file) [{success}] Time taken:", time.time() - begin)
9297
begin = time.time()
9398
result = lib.convertToJson(tree_id)
94-
print(f"JSON (raw) [{len(result)}] Time taken:", time.time() - begin)
95-
96-
# Confirm it complies with UTF-8
9799
check_decode(result, 'JSON (raw)')
100+
print(f"JSON (raw) [{len(result)}] Time taken:", time.time() - begin)
98101

99102
scene_info = lib.getSceneInfo(tree_id)
100103
if scene_info:
@@ -117,23 +120,33 @@ def check_decode(raw: bytes, name: str):
117120

118121
begin = time.time()
119122
success = lib.textToMdFile(renderer_id, md_output_path.encode())
123+
if not success:
124+
exit(-1)
120125
print(f"MD (file) [{success}] Time taken:", time.time() - begin)
121126
begin = time.time()
122127
result: bytes = lib.textToMd(renderer_id)
128+
check_decode(result, 'MD (raw)')
123129
print(f"MD (raw) [{len(result)}] Time taken:", time.time() - begin)
124130

125-
# Confirm it complies with UTF-8
131+
begin = time.time()
132+
success = lib.textToTxtFile(renderer_id, txt_output_path.encode())
133+
if not success:
134+
exit(-1)
135+
print(f"TXT (file) [{success}] Time taken:", time.time() - begin)
136+
begin = time.time()
137+
result: bytes = lib.textToTxt(renderer_id)
126138
check_decode(result, 'MD (raw)')
139+
print(f"TXT (raw) [{len(result)}] Time taken:", time.time() - begin)
127140

128141
begin = time.time()
129142
success = lib.textToHtmlFile(renderer_id, html_output_path.encode())
143+
if not success:
144+
exit(-1)
130145
print(f"HTML (file) [{success}] Time taken:", time.time() - begin)
131146
begin = time.time()
132147
result = lib.textToHtml(renderer_id)
133-
print(f"HTML (raw) [{len(result)}] Time taken:", time.time() - begin)
134-
135-
# Confirm it complies with UTF-8
136148
check_decode(result, 'HTML (raw)')
149+
print(f"HTML (raw) [{len(result)}] Time taken:", time.time() - begin)
137150

138151
begin = time.time()
139152
size_of_renderer = lib.destroyRenderer(renderer_id)

0 commit comments

Comments
 (0)