diff --git a/pyproject.toml b/pyproject.toml index 210fba6..41f2ef9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "hatchling.build" [project] name = "refinedoc" -version = "1.0.0" +version = "1.0.1" authors = [ { name="Théo NARDIN", email="theo.nardin@learningplanetinstitute.org" }, ] diff --git a/src/refinedoc/refined_document.py b/src/refinedoc/refined_document.py index b095d92..d304ff0 100644 --- a/src/refinedoc/refined_document.py +++ b/src/refinedoc/refined_document.py @@ -43,8 +43,14 @@ def __init__(self, content: list[list[str]], ratio_speed: int = 1, win: int = 8) raise ValueError(f"Speed must be between 1 and 3: {ratio_speed}") self._processed_body: list[list[str]] = content # Initialize body field - self._processed_headers: list[list[str]] | None = None - self._processed_footers: list[list[str]] | None = None + + "The content provided has only one page. Headers and footers will be set to empty lists." + ) + self._processed_headers = [[]] + self._processed_footers = [[]] + else: + self._processed_headers: list[list[str]] | None = None + self._processed_footers: list[list[str]] | None = None self.win = win diff --git a/tests/test_refined_document.py b/tests/test_refined_document.py index 49f9cfc..a1b148a 100644 --- a/tests/test_refined_document.py +++ b/tests/test_refined_document.py @@ -404,3 +404,57 @@ def test_various_qty_lines2(self): self.assertListEqual(h_dr, h_ref) self.assertListEqual(f_dr, f_ref) self.assertListEqual(b_dr, b_ref) + + def test_empty_document(self): + document = [] + + rd = RefinedDocument(content=document) + + h_dr = rd.headers + f_dr = rd.footers + b_dr = rd.body + + h_ref = [] + + f_ref = [] + + b_ref = [] + + self.assertListEqual(h_dr, h_ref) + self.assertListEqual(f_dr, f_ref) + self.assertListEqual(b_dr, b_ref) + + def test_single_page_document(self): + document = [ + [ + "header 1", + "subheader 1", + "lorem ipsum dolor sit amet", + "consectetur adipiscing elit", + "footer 1", + ] + ] + + rd = RefinedDocument(content=document) + + h_dr = rd.headers + f_dr = rd.footers + b_dr = rd.body + + h_ref = [[]] + + f_ref = [[]] + + b_ref = [ + [ + "header 1", + "subheader 1", + "lorem ipsum dolor sit amet", + "consectetur adipiscing elit", + "footer 1", + ] + ] + + self.assertListEqual(h_dr, h_ref) + self.assertListEqual(f_dr, f_ref) + self.assertListEqual(b_dr, b_ref)