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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
import os
import base64
import spur
import tempman
from .testing import assert_equal, generate_test_path
_local = spur.LocalShell()
def test_html_is_printed_to_stdout_if_output_file_is_not_set():
docx_path = generate_test_path("single-paragraph.docx")
result = _local.run(["mammoth", docx_path])
assert_equal(b"", result.stderr_output)
assert_equal(b"<p>Walking on imported air</p>", result.output)
def test_html_is_written_to_file_if_output_file_is_set():
with tempman.create_temp_dir() as temp_dir:
output_path = os.path.join(temp_dir.path, "output.html")
docx_path = generate_test_path("single-paragraph.docx")
result = _local.run(["mammoth", docx_path, output_path])
assert_equal(b"", result.stderr_output)
assert_equal(b"", result.output)
with open(output_path) as output_file:
assert_equal("<p>Walking on imported air</p>", output_file.read())
_image_base_64 = b"iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAIAAAACUFjqAAAAAXNSR0IArs4c6QAAAAlwSFlzAAAOvgAADr4B6kKxwAAAABNJREFUKFNj/M+ADzDhlWUYqdIAQSwBE8U+X40AAAAASUVORK5CYII="
def test_inline_images_are_included_in_output_if_writing_to_single_file():
docx_path = generate_test_path("tiny-picture.docx")
result = _local.run(["mammoth", docx_path])
assert_equal(b"""<p><img src="data:image/png;base64,""" + _image_base_64 + b"""" /></p>""", result.output)
def test_images_are_written_to_separate_files_if_output_dir_is_set():
with tempman.create_temp_dir() as temp_dir:
output_path = os.path.join(temp_dir.path, "tiny-picture.html")
image_path = os.path.join(temp_dir.path, "1.png")
docx_path = generate_test_path("tiny-picture.docx")
result = _local.run(["mammoth", docx_path, "--output-dir", temp_dir.path])
assert_equal(b"", result.stderr_output)
assert_equal(b"", result.output)
with open(output_path) as output_file:
assert_equal("""<p><img src="1.png" /></p>""", output_file.read())
with open(image_path, "rb") as image_file:
assert_equal(_image_base_64, base64.b64encode(image_file.read()))
def test_style_map_is_used_if_set():
with tempman.create_temp_dir() as temp_dir:
docx_path = generate_test_path("single-paragraph.docx")
style_map_path = os.path.join(temp_dir.path, "style-map")
with open(style_map_path, "w") as style_map_file:
style_map_file.write("p => span:fresh")
result = _local.run(["mammoth", docx_path, "--style-map", style_map_path])
assert_equal(b"", result.stderr_output)
assert_equal(b"<span>Walking on imported air</span>", result.output)
def test_output_format_markdown_option_generates_markdown_output():
docx_path = generate_test_path("single-paragraph.docx")
result = _local.run(["mammoth", docx_path, "--output-format=markdown"])
assert_equal(b"", result.stderr_output)
assert_equal(b"Walking on imported air\n\n", result.output)
|
