Add sample wrapper (#79)

* add `sample` wrapper

* [pre-commit.ci lite] apply automatic fixes

---------

Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>

Author: jenna-tomkinson

.pre-commit-config.yaml

@@ -16,7 +16,7 @@ repos:
     -   id: check-added-large-files
     -   id: detect-private-key
 -   repo: https://github.com/python-poetry/poetry
-    rev: "2.0.1"
+    rev: "2.1.1"
     hooks:
     -   id: poetry-check
 -   repo: https://github.com/tox-dev/pyproject-fmt

media/coverage-badge.svg

@@ -320,7 +320,7 @@ def _wrap_methods(self) -> None:
         """
 
         # list of methods by name from Pandas DataFrame class
-        methods_to_wrap = ["head", "tail", "sort_values"]
+        methods_to_wrap = ["head", "tail", "sort_values", "sample"]
 
         # set the wrapped method for the class instance
         for method_name in methods_to_wrap:

src/cytodataframe/frame.py

@@ -221,3 +221,4 @@ def test_return_cytodataframe(cytotable_NF1_data_parquet_shrunken: str):
     assert isinstance(cdf.head(), CytoDataFrame)
     assert isinstance(cdf.tail(), CytoDataFrame)
     assert isinstance(cdf.sort_values(by="Metadata_ImageNumber"), CytoDataFrame)
+    assert isinstance(cdf.sample(n=5), CytoDataFrame)