test: Add tests for formatter sad paths
This commit is contained in:
parent
f67ac8cdb3
commit
c9736a5f32
1 changed files with 28 additions and 7 deletions
|
@ -6,6 +6,7 @@ from papis_extract.formatter import (
|
||||||
format_csv,
|
format_csv,
|
||||||
format_markdown,
|
format_markdown,
|
||||||
format_markdown_atx,
|
format_markdown_atx,
|
||||||
|
format_markdown_setext,
|
||||||
)
|
)
|
||||||
|
|
||||||
an_doc: AnnotatedDocument = AnnotatedDocument(
|
an_doc: AnnotatedDocument = AnnotatedDocument(
|
||||||
|
@ -15,12 +16,7 @@ an_doc: AnnotatedDocument = AnnotatedDocument(
|
||||||
Annotation("myfile.pdf", text="my second text", content="with note"),
|
Annotation("myfile.pdf", text="my second text", content="with note"),
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
md_default_output = """============== ---------------
|
||||||
|
|
||||||
def test_markdown_default():
|
|
||||||
fmt = format_markdown
|
|
||||||
assert fmt([an_doc]) == (
|
|
||||||
"""============== ---------------
|
|
||||||
document-title - document-author
|
document-title - document-author
|
||||||
============== ---------------
|
============== ---------------
|
||||||
|
|
||||||
|
@ -28,7 +24,11 @@ document-title - document-author
|
||||||
|
|
||||||
> my second text
|
> my second text
|
||||||
NOTE: with note"""
|
NOTE: with note"""
|
||||||
)
|
|
||||||
|
|
||||||
|
def test_markdown_default():
|
||||||
|
fmt = format_markdown
|
||||||
|
assert fmt([an_doc]) == md_default_output
|
||||||
|
|
||||||
|
|
||||||
def test_markdown_atx():
|
def test_markdown_atx():
|
||||||
|
@ -43,6 +43,11 @@ def test_markdown_atx():
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_markdown_setext():
|
||||||
|
fmt = format_markdown_setext
|
||||||
|
assert fmt([an_doc]) == md_default_output
|
||||||
|
|
||||||
|
|
||||||
def test_count_default():
|
def test_count_default():
|
||||||
fmt = format_count
|
fmt = format_count
|
||||||
assert fmt([an_doc]) == ("""document-author - document-title: 2""")
|
assert fmt([an_doc]) == ("""document-author - document-title: 2""")
|
||||||
|
@ -57,3 +62,19 @@ def test_csv_default():
|
||||||
'Highlight,,0,"my second text","with note","document-author",'
|
'Highlight,,0,"my second text","with note","document-author",'
|
||||||
'"document-title","","myfile.pdf"'
|
'"document-title","","myfile.pdf"'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# sadpath - no annotations contained for each format
|
||||||
|
def test_markdown_no_annotations():
|
||||||
|
d: AnnotatedDocument = AnnotatedDocument(Document(data={}), [])
|
||||||
|
assert format_markdown([d]) == ""
|
||||||
|
|
||||||
|
|
||||||
|
def test_count_no_annotations():
|
||||||
|
d: AnnotatedDocument = AnnotatedDocument(Document(data={}), [])
|
||||||
|
assert format_count([d]) == ""
|
||||||
|
|
||||||
|
|
||||||
|
def test_csv_no_annotations():
|
||||||
|
d: AnnotatedDocument = AnnotatedDocument(Document(data={}), [])
|
||||||
|
assert format_csv([d]) == "type,tag,page,quote,note,author,title,ref,file"
|
||||||
|
|
Loading…
Reference in a new issue