adding tests and docs to head and tail

groundgametexas · Feb 24, 2024 · a885d11 · a885d11
1 parent 2a4290d
commit a885d11
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/docs/table.rst b/docs/table.rst
@@ -156,6 +156,10 @@ of commonly used methods. The full list can be found in the API section.
 
     * - Method
       - Description
+    * - :py:meth:`~parsons.etl.etl.ETL.head`
+      - Get the first n rows of a table
+    * - :py:meth:`~parsons.etl.etl.ETL.tail`
+      - Get the last n rows of a table
     * - :py:meth:`~parsons.etl.etl.ETL.add_column`
       - Add a column
     * - :py:meth:`~parsons.etl.etl.ETL.remove_column`

diff --git a/test/test_etl.py b/test/test_etl.py
@@ -997,3 +997,15 @@ def test_deduplicate(self):
         tbl_expected = Table([["a", "b", "c"], [1, 2, 3], [1, 3, 2], [2, 3, 4]])
         tbl.deduplicate(["a", "b"])
         assert_matching_tables(tbl_expected, tbl)
+
+    def test_head(self):
+        tbl = Table([["a", "b"], [1, 2], [3, 4], [5, 6], [7, 8], [9, 10]])
+        tbl_expected = Table([["a", "b"], [1, 2], [3, 4]])
+        tbl.head(2)
+        assert_matching_tables(tbl_expected, tbl)
+
+    def test_tail(self):
+        tbl = Table([["a", "b"], [1, 2], [3, 4], [5, 6], [7, 8], [9, 10]])
+        tbl_expected = Table([["a", "b"], [7, 8], [9, 10]])
+        tbl.tail(2)
+        assert_matching_tables(tbl_expected, tbl)