Fix/no candidates

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" },
 ]

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
 

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)