Rename _reorganize_ipynb... to ensure_ipynb_compatibility

Author: ma-sadeghi

src/pytest_split/ipynb_compatibility.py

@@ -8,7 +8,7 @@
     from pytest_split.algorithms import TestGroup
 
 
-def _reorganize_broken_up_ipynbs(group: "TestGroup", items: "List[nodes.Item]") -> None:
+def ensure_ipynb_compatibility(group: "TestGroup", items: "List[nodes.Item]") -> None:
     """
     Ensures that group doesn't contain partial IPy notebook cells.
 
@@ -44,7 +44,9 @@ def _reorganize_broken_up_ipynbs(group: "TestGroup", items: "List[nodes.Item]")
                 group.selected.append(item)
 
 
-def _find_sibiling_ipynb_cells(ipynb_node_id: str, item_node_ids: "List[str]") -> "List[str]":
+def _find_sibiling_ipynb_cells(
+    ipynb_node_id: str, item_node_ids: "List[str]"
+) -> "List[str]":
     """
     Returns all sibiling IPyNb cells given an IPyNb cell nodeid.
     """

src/pytest_split/plugin.py

@@ -7,7 +7,7 @@
 from _pytest.reports import TestReport
 
 from pytest_split import algorithms
-from pytest_split.ipynb_compatibility import _reorganize_broken_up_ipynbs
+from pytest_split.ipynb_compatibility import ensure_ipynb_compatibility
 
 if TYPE_CHECKING:
     from typing import Dict, List, Optional, Union
@@ -167,7 +167,7 @@ def pytest_collection_modifyitems(
         groups = algo(splits, items, self.cached_durations)
         group = groups[group_idx - 1]
 
-        _reorganize_broken_up_ipynbs(group, items)
+        ensure_ipynb_compatibility(group, items)
 
         items[:] = group.selected
         config.hook.pytest_deselected(items=group.deselected)

tests/test_ipynb.py

@@ -3,14 +3,14 @@
 import pytest
 
 from pytest_split.algorithms import Algorithms
-from pytest_split.ipynb_compatibility import _reorganize_broken_up_ipynbs
+from pytest_split.ipynb_compatibility import ensure_ipynb_compatibility
 
 item = namedtuple("item", "nodeid")
 
 
 class TestIPyNb:
     @pytest.mark.parametrize("algo_name", ["duration_based_chunks"])
-    def test__reorganize_broken_up_ipynbs(self, algo_name):
+    def test_ensure_ipynb_compatibility(self, algo_name):
         durations = {
             "temp/nbs/test_1.ipynb::Cell 0": 1,
             "temp/nbs/test_1.ipynb::Cell 1": 1,
@@ -54,7 +54,7 @@ def test__reorganize_broken_up_ipynbs(self, algo_name):
             item(nodeid="temp/nbs/test_4.ipynb::Cell 2"),
         ]
 
-        _reorganize_broken_up_ipynbs(groups[0], items)
+        ensure_ipynb_compatibility(groups[0], items)
         assert groups[0].selected == [
             item(nodeid="temp/nbs/test_1.ipynb::Cell 0"),
             item(nodeid="temp/nbs/test_1.ipynb::Cell 1"),
@@ -65,7 +65,7 @@ def test__reorganize_broken_up_ipynbs(self, algo_name):
             item(nodeid="temp/nbs/test_2.ipynb::Cell 3"),
         ]
 
-        _reorganize_broken_up_ipynbs(groups[1], items)
+        ensure_ipynb_compatibility(groups[1], items)
         assert groups[1].selected == [
             item(nodeid="temp/nbs/test_3.ipynb::Cell 0"),
             item(nodeid="temp/nbs/test_3.ipynb::Cell 1"),
@@ -74,7 +74,7 @@ def test__reorganize_broken_up_ipynbs(self, algo_name):
             item(nodeid="temp/nbs/test_3.ipynb::Cell 4"),
         ]
 
-        _reorganize_broken_up_ipynbs(groups[2], items)
+        ensure_ipynb_compatibility(groups[2], items)
         assert groups[2].selected == [
             item(nodeid="temp/nbs/test_4.ipynb::Cell 0"),
             item(nodeid="temp/nbs/test_4.ipynb::Cell 1"),