refactor: store file in position, rename filename to file_name
Signed-off-by: Konstantin Fickel <mail@konstantinfickel.de>
This commit is contained in:
parent
d5b1541436
commit
79095bad4a
8 changed files with 27 additions and 24 deletions
|
|
@ -16,12 +16,12 @@ class TestAttachMarkdown:
|
|||
def test_attach_markdown_with_shard(self):
|
||||
markdown_text = "Hello World\n\nThis is a test."
|
||||
shard = Shard(start_line=1, end_line=3)
|
||||
stream_file = StreamFile(filename=self.file_name, shard=shard)
|
||||
stream_file = StreamFile(file_name=self.file_name, shard=shard)
|
||||
|
||||
result = attach_markdown(stream_file, markdown_text)
|
||||
|
||||
assert result == StreamFileWithMarkdown(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=ShardWithMarkdown(
|
||||
start_line=1,
|
||||
end_line=3,
|
||||
|
|
@ -33,11 +33,11 @@ class TestAttachMarkdown:
|
|||
)
|
||||
|
||||
def test_attach_markdown_without_shard(self):
|
||||
stream_file = StreamFile(filename=self.file_name, shard=None)
|
||||
stream_file = StreamFile(file_name=self.file_name, shard=None)
|
||||
|
||||
result = attach_markdown(stream_file, "Some markdown text")
|
||||
|
||||
assert result == StreamFileWithMarkdown(filename=self.file_name, shard=None)
|
||||
assert result == StreamFileWithMarkdown(file_name=self.file_name, shard=None)
|
||||
|
||||
def test_attach_markdown_with_nested_shards(self):
|
||||
markdown_text = "Header\n\n@Marker1 Content 1\n\n@Marker2 Content 2"
|
||||
|
|
@ -49,11 +49,11 @@ class TestAttachMarkdown:
|
|||
Shard(markers=["Marker2"], start_line=5, end_line=5),
|
||||
],
|
||||
)
|
||||
stream_file = StreamFile(filename=self.file_name, shard=shard)
|
||||
stream_file = StreamFile(file_name=self.file_name, shard=shard)
|
||||
|
||||
result = attach_markdown(stream_file, markdown_text)
|
||||
|
||||
assert result.filename == self.file_name
|
||||
assert result.file_name == self.file_name
|
||||
assert result.shard is not None
|
||||
assert result.shard.start_line == 1
|
||||
assert result.shard.end_line == 5
|
||||
|
|
|
|||
|
|
@ -10,13 +10,13 @@ class TestParseProcess:
|
|||
|
||||
def test_parse_empty_file(self):
|
||||
assert parse_markdown_file(self.file_name, "") == StreamFile(
|
||||
filename=self.file_name, shard=Shard(start_line=1, end_line=1)
|
||||
file_name=self.file_name, shard=Shard(start_line=1, end_line=1)
|
||||
)
|
||||
|
||||
def test_parse_basic_one_line_file(self):
|
||||
test_file = "Hello World"
|
||||
assert parse_markdown_file(self.file_name, test_file) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
start_line=1,
|
||||
end_line=1,
|
||||
|
|
@ -26,7 +26,7 @@ class TestParseProcess:
|
|||
def test_parse_basic_multi_line_file(self):
|
||||
test_file = "Hello World\n\nHello again!"
|
||||
assert parse_markdown_file(self.file_name, test_file) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
start_line=1,
|
||||
end_line=3,
|
||||
|
|
@ -36,7 +36,7 @@ class TestParseProcess:
|
|||
def test_parse_single_line_with_tag(self):
|
||||
test_file = "@Tag Hello World"
|
||||
assert parse_markdown_file(self.file_name, test_file) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
markers=["Tag"],
|
||||
start_line=1,
|
||||
|
|
@ -47,7 +47,7 @@ class TestParseProcess:
|
|||
def test_parse_single_line_with_two_tags(self):
|
||||
test_file = "@Marker1 @Marker2 Hello World"
|
||||
assert parse_markdown_file(self.file_name, test_file) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
markers=["Marker1", "Marker2"],
|
||||
start_line=1,
|
||||
|
|
@ -58,7 +58,7 @@ class TestParseProcess:
|
|||
def test_parse_single_line_with_two_tags_and_misplaced_tag(self):
|
||||
test_file = "@Tag1 @Tag2 Hello World @Tag3"
|
||||
assert parse_markdown_file(self.file_name, test_file) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
markers=["Tag1", "Tag2"],
|
||||
tags=["Tag3"],
|
||||
|
|
@ -71,7 +71,7 @@ class TestParseProcess:
|
|||
file_text = "Hello World!\n\n@Tag1 Block 1\n\n@Tag2 Block 2"
|
||||
|
||||
assert parse_markdown_file(self.file_name, file_text) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
start_line=1,
|
||||
end_line=5,
|
||||
|
|
@ -114,7 +114,7 @@ class TestParseProcess:
|
|||
file_text = "# Heading @Tag1\n\n## @Marker1 Subheading @Tag2\n\n# Heading @Tag3"
|
||||
|
||||
assert parse_markdown_file(self.file_name, file_text) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
start_line=1,
|
||||
end_line=5,
|
||||
|
|
@ -141,7 +141,7 @@ class TestParseProcess:
|
|||
file_text = "# @Marker1 Heading @Tag1\n\n## Subheading @Tag2"
|
||||
|
||||
assert parse_markdown_file(self.file_name, file_text) == StreamFile(
|
||||
filename=self.file_name,
|
||||
file_name=self.file_name,
|
||||
shard=Shard(
|
||||
markers=["Marker1"],
|
||||
tags=["Tag1", "Tag2"],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue