autogaita 1.5.2__py3-none-any.whl → 1.5.4__py3-none-any.whl

tests/test_common2D_unit_2_sc_extraction.py

@@ -1,367 +0,0 @@
-from autogaita.dlc.dlc_main import dlc
-from autogaita.common2D.common2D_1_preparation import some_prep
-from autogaita.common2D.common2D_2_sc_extraction import extract_stepcycles
-from autogaita.common2D.common2D_utils import (
-    check_cycle_out_of_bounds,
-    check_cycle_duplicates,
-    check_cycle_order,
-    check_differing_angle_joint_coords,
-    check_tracking_xy_thresholds,
-    check_tracking_SLEAP_nans,
-)
-from hypothesis import given, strategies as st, settings, HealthCheck
-import os
-import numpy as np
-import pytest
-
-
-# %%................................  fixtures  ........................................
-@pytest.fixture
-def extract_data_using_some_prep(extract_info, extract_folderinfo, extract_cfg):
-    data = some_prep("DLC", extract_info, extract_folderinfo, extract_cfg)
-    return data
-
-
-@pytest.fixture
-def extract_info(tmp_path):
-    info = {}
-    info["mouse_num"] = 15
-    info["run_num"] = 3
-    info["name"] = "ID " + str(info["mouse_num"]) + " - Run " + str(info["run_num"])
-    info["results_dir"] = os.path.join(tmp_path, info["name"])
-    return info
-
-
-@pytest.fixture
-def extract_folderinfo():
-    folderinfo = {}
-    folderinfo["root_dir"] = "tests/test_data/dlc_data"
-    folderinfo["sctable_filename"] = (
-        "correct_annotation_table.xlsx"  # has to be an excel file
-    )
-    folderinfo["data_string"] = "SIMINewOct"
-    folderinfo["beam_string"] = "BeamTraining"
-    folderinfo["premouse_string"] = "Mouse"
-    folderinfo["postmouse_string"] = "25mm"
-    folderinfo["prerun_string"] = "run"
-    folderinfo["postrun_string"] = "6DLC"
-    return folderinfo
-
-
-@pytest.fixture
-def extract_cfg():
-    cfg = {}
-    cfg["sampling_rate"] = 100
-    cfg["subtract_beam"] = True
-    cfg["dont_show_plots"] = True
-    cfg["convert_to_mm"] = True
-    cfg["pixel_to_mm_ratio"] = 3.76
-    cfg["x_sc_broken_threshold"] = 200  # optional cfg
-    cfg["y_sc_broken_threshold"] = 50
-    cfg["x_acceleration"] = True
-    cfg["angular_acceleration"] = True
-    cfg["save_to_xls"] = True
-    cfg["bin_num"] = 25
-    cfg["plot_SE"] = True
-    cfg["standardise_y_at_SC_level"] = False
-    cfg["standardise_y_to_a_joint"] = True
-    cfg["y_standardisation_joint"] = ["Knee"]  # "Hind paw tao"]
-    cfg["plot_joint_number"] = 3
-    cfg["color_palette"] = "viridis"
-    cfg["legend_outside"] = True
-    cfg["invert_y_axis"] = True
-    cfg["flip_gait_direction"] = True
-    cfg["analyse_average_x"] = True
-    cfg["standardise_x_coordinates"] = True
-    cfg["x_standardisation_joint"] = ["Hind paw tao"]
-    cfg["coordinate_standardisation_xls"] = ""
-    cfg["hind_joints"] = ["Hind paw tao", "Ankle", "Knee", "Hip", "Iliac Crest"]
-    cfg["fore_joints"] = [
-        "Front paw tao ",
-        "Wrist ",
-        "Elbow ",
-        "Lower Shoulder ",
-        "Upper Shoulder ",
-    ]
-    cfg["beam_col_left"] = ["BeamLeft"]  # BEAM_COL_LEFT & _RIGHT must be lists of len=1
-    cfg["beam_col_right"] = ["BeamRight"]
-    cfg["beam_hind_jointadd"] = ["Tail base ", "Tail center ", "Tail tip "]
-    cfg["beam_fore_jointadd"] = ["Nose ", "Ear base "]
-    cfg["angles"] = {
-        "name": ["Ankle ", "Knee ", "Hip "],
-        "lower_joint": ["Hind paw tao ", "Ankle ", "Knee "],
-        "upper_joint": ["Knee ", "Hip ", "Iliac Crest "],
-    }
-    return cfg
-
-
-# %%..............................  test golden path  ..................................
-
-
-def test_golden_path_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    expected_cycles = [[284, 317], [318, 359], [413, 441]]
-    assert (
-        extract_stepcycles(
-            "DLC",
-            extract_data_using_some_prep,
-            extract_info,
-            extract_folderinfo,
-            extract_cfg,
-        )
-        == expected_cycles
-    )
-
-
-# %%...................  test Annotation Table checks & handle_issues  .................
-
-
-def test_file_not_found_error_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_folderinfo["root_dir"] = ""
-    with pytest.raises(FileNotFoundError) as excinfo:
-        extract_stepcycles(
-            "DLC",
-            extract_data_using_some_prep,
-            extract_info,
-            extract_folderinfo,
-            extract_cfg,
-        )
-    assert "No Annotation Table found!" in str(excinfo.value)
-
-
-def test_handle_issues_1_all_SCs_invalid_because_all_cycles_empty_in_dlc(
-    extract_info, extract_folderinfo, extract_cfg
-):
-    # call dlc and not extract_stepcycles since handle_issues call we test is in dlc
-    extract_folderinfo["root_dir"] = os.path.join(
-        extract_folderinfo["root_dir"], "flawed_files"
-    )
-    extract_folderinfo["sctable_filename"] = "flawed_table_all_ID_15_SCs_invalid"
-    dlc(extract_info, extract_folderinfo, extract_cfg)
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert ("Skipped since all SCs invalid!" in content) & (
-        "not in data/video" in content
-    )
-
-
-def test_handle_issues_2_no_scs_of_given_mouse_and_run_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_info["mouse_num"] = 12
-    extract_info["run_num"] = 1
-    extract_stepcycles(
-        "DLC",
-        extract_data_using_some_prep,
-        extract_info,
-        extract_folderinfo,
-        extract_cfg,
-    )
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "Skipped since no SCs in Annotation Table!" in content
-
-
-def test_handle_issues_2_wrong_run_number_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_info["run_num"] = 123456789101112
-    extract_stepcycles(
-        "DLC",
-        extract_data_using_some_prep,
-        extract_info,
-        extract_folderinfo,
-        extract_cfg,
-    )
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "Skipped since no SCs in Annotation Table!" in content
-
-
-def test_handle_issues_3_wrong_mouse_number_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_info["mouse_num"] = 123456789101112
-    extract_stepcycles(
-        "DLC",
-        extract_data_using_some_prep,
-        extract_info,
-        extract_folderinfo,
-        extract_cfg,
-    )
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "ID not in Annotation Table!" in content
-
-
-def test_handle_issues_4_bad_annotation_table_columns_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_folderinfo["root_dir"] = os.path.join(
-        extract_folderinfo["root_dir"], "flawed_files"
-    )
-    extract_folderinfo["sctable_filename"] = "flawed_table_bad_column_names_table"
-    extract_stepcycles(
-        "DLC",
-        extract_data_using_some_prep,
-        extract_info,
-        extract_folderinfo,
-        extract_cfg,
-    )
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "Annotation Table's Column Names are wrong!" in content
-
-
-def test_handle_issues_5_double_ID_in_annotation_table_in_extract_stepcycles(
-    extract_data_using_some_prep, extract_info, extract_folderinfo, extract_cfg
-):
-    extract_folderinfo["root_dir"] = os.path.join(
-        extract_folderinfo["root_dir"], "flawed_files"
-    )
-    extract_folderinfo["sctable_filename"] = "flawed_table_double_ID_15_table"
-    extract_stepcycles(
-        "DLC",
-        extract_data_using_some_prep,
-        extract_info,
-        extract_folderinfo,
-        extract_cfg,
-    )
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "ID found more than once in Annotation Table!" in content
-
-
-# .....................  test clean all_cycles local functions  ........................
-
-
-@given(
-    all_cycles=st.lists(
-        st.lists(
-            st.one_of(st.integers(), st.floats(), st.text()), min_size=2, max_size=2
-        )
-    )
-)
-def test_clean_cycles_1a_cycle_out_of_bounds_in_extract_stepcycles(all_cycles):
-    all_cycles = check_cycle_out_of_bounds(all_cycles)
-    flat_cycles = flatten_all_cycles(all_cycles)
-    if all_cycles:  # can be None
-        assert all(isinstance(idx, (int, np.integer)) for idx in flat_cycles)
-
-
-# Note for following cases that within extract_stepcycles a check for cycle-idxs being in data.index assigns all_cycles[s] to [None, None] - so we have to use that here
-cases = (
-    (
-        [[1, 100], [None, None], [200, 300]],
-        [[1, 100], [200, 300]],
-    ),
-    ([[None, None], [None, None], [None, None]], None),
-)  # fmt: skip
-@pytest.mark.parametrize("all_cycles, expected_cycles", cases)
-def test_clean_cycles_1b_cycle_out_of_bounds_in_extract_stepcycles(
-    all_cycles, expected_cycles
-):
-    assert expected_cycles == check_cycle_out_of_bounds(all_cycles)
-
-
-cases = (
-    (
-        [[11, 12], [12, 14], [14, 110], [110, 210]],
-        [[11, 12], [13, 14], [15, 110], [111, 210]],
-    ),
-    ([[1, 2], [2, 3], [3, 4], [4, 5]], [[1, 2], [3, 3], [4, 4], [5, 5]]),
-)  # fmt: skip
-@pytest.mark.parametrize("all_cycles, expected_cycles", cases)
-def test_clean_cycles_2_cycle_duplicates_in_extract_stepcycles(
-    all_cycles, expected_cycles
-):
-    assert expected_cycles == check_cycle_duplicates(all_cycles)
-
-
-@given(all_cycles=st.lists(st.lists(st.integers(), min_size=2, max_size=2)))
-@settings(suppress_health_check=[HealthCheck.function_scoped_fixture])
-def test_clean_cycles_3_cycle_order(all_cycles, extract_info):
-    """Note here that we manually create results_dir's tmp_path if required bc. it's usually created in dlc somewhere and that hypothesis complains without the if not condition bc. the path keeps being created for new cases I think!"""
-    if not os.path.exists(extract_info["results_dir"]):
-        os.makedirs(extract_info["results_dir"])
-    all_cycles = check_cycle_order(all_cycles, extract_info)
-    flat_cycles = flatten_all_cycles(all_cycles)
-    if all_cycles:  # can be None
-        assert flat_cycles == sorted(flat_cycles)
-
-
-def test_clean_cycles_4_differing_angle_joint_coords(
-    extract_data_using_some_prep, extract_info, extract_cfg
-):
-    data = extract_data_using_some_prep.copy()
-    all_cycles = [[111, 222], [333, 444]]
-    data.loc[123, "Hind paw tao x"] = 5.5
-    data.loc[123, "Hind paw tao y"] = 2.1
-    data.loc[123, "Ankle x"] = 5.5
-    data.loc[123, "Ankle y"] = 2.1
-    all_cycles = check_differing_angle_joint_coords(
-        all_cycles, data, extract_info, extract_cfg
-    )
-    assert all_cycles == [[333, 444]]
-    with open(os.path.join(extract_info["results_dir"], "Issues.txt")) as f:
-        content = f.read()
-    assert "SC #1" in content
-    assert "Lower joint: Hind paw tao" in content
-    data.loc[345, "Hip x"] = 10.1
-    data.loc[345, "Hip y"] = 0.333
-    data.loc[345, "Iliac Crest x"] = 10.1
-    data.loc[345, "Iliac Crest y"] = 0.333
-    assert (
-        check_differing_angle_joint_coords(all_cycles, data, extract_info, extract_cfg)
-        is None
-    )
-
-
-def test_clean_cycles_5_DLC_tracking(
-    extract_data_using_some_prep, extract_info, extract_cfg
-):
-    """Note that we know that very early on (2-150) the mouse was not in the frame yet so DLC is broken and these SCs will be excluded. The other 3 of case 2 are the correcty SCs of this ID/run of which None should be excluded!"""
-    all_cycles_of_the_two_cases = (
-        [[2, 50], [52, 100], [102, 150]],
-        [[2, 50], [52, 100], [102, 150], [284, 317], [318, 359], [413, 441]],
-    )
-    expected_cycles = (None, [[284, 317], [318, 359], [413, 441]])
-    for c, this_cases_all_cycles in enumerate(all_cycles_of_the_two_cases):
-        assert expected_cycles[c] == check_tracking_xy_thresholds(
-            this_cases_all_cycles,
-            extract_data_using_some_prep,
-            extract_info,
-            extract_cfg,
-        )
-
-
-def test_clean_cycles_6_SLEAP_nan_removals(
-    extract_data_using_some_prep, extract_info, extract_cfg
-):
-    """We just pretend that this is SLEAP data - whatever"""
-    extract_cfg["joints"] = ["Hind paw tao ", "Ankle "]
-    extract_cfg["angles"] = {
-        "name": ["Elbow "],
-        "lower_joint": ["Wrist "],
-        "upper_joint": ["Lower Shoulder "],
-    }
-    data = extract_data_using_some_prep.copy()
-    all_cycles = [[111, 222], [333, 444], [555, 666]]
-    data.loc[123, "Hind paw tao y"] = np.nan
-    data.loc[444, "Elbow x"] = np.nan
-    clean_cycles = check_tracking_SLEAP_nans(
-        all_cycles,
-        data,
-        extract_info,
-        extract_cfg,
-    )
-    assert clean_cycles == [[555, 666]]
-
-
-# ...............................  helper functions  ...................................
-def flatten_all_cycles(all_cycles):
-    if all_cycles:
-        return [idx for cycle in all_cycles for idx in cycle]

tests/test_common2D_unit_3_analysis.py

@@ -1,245 +0,0 @@
-from autogaita.common2D.common2D_1_preparation import some_prep
-from autogaita.common2D.common2D_2_sc_extraction import extract_stepcycles
-from autogaita.common2D.common2D_3_analysis import (
-    analyse_and_export_stepcycles,
-    add_step_separators,
-    add_angles,
-    add_x_velocities,
-    add_angular_velocities,
-)
-from hypothesis import given
-import hypothesis.strategies as st
-from hypothesis.extra.numpy import arrays
-import os
-import numpy as np
-import pandas as pd
-import pandas.testing as pdt
-import pytest
-
-
-# %%................................  fixtures  ........................................
-@pytest.fixture
-def extract_info(tmp_path):
-    info = {}
-    info["mouse_num"] = 15
-    info["run_num"] = 3
-    info["name"] = "ID " + str(info["mouse_num"]) + " - Run " + str(info["run_num"])
-    info["results_dir"] = os.path.join(tmp_path, info["name"])
-    return info
-
-
-@pytest.fixture
-def extract_folderinfo():
-    folderinfo = {}
-    folderinfo["root_dir"] = "tests/test_data/dlc_data"
-    folderinfo["sctable_filename"] = (
-        "correct_annotation_table.xlsx"  # has to be an excel file
-    )
-    folderinfo["data_string"] = "SIMINewOct"
-    folderinfo["beam_string"] = "BeamTraining"
-    folderinfo["premouse_string"] = "Mouse"
-    folderinfo["postmouse_string"] = "25mm"
-    folderinfo["prerun_string"] = "run"
-    folderinfo["postrun_string"] = "6DLC"
-    return folderinfo
-
-
-@pytest.fixture
-def extract_cfg():
-    cfg = {}
-    cfg["sampling_rate"] = 100
-    cfg["subtract_beam"] = True
-    cfg["dont_show_plots"] = True
-    cfg["convert_to_mm"] = True
-    cfg["pixel_to_mm_ratio"] = 3.76
-    cfg["x_sc_broken_threshold"] = 200  # optional cfg
-    cfg["y_sc_broken_threshold"] = 50
-    cfg["x_acceleration"] = True
-    cfg["angular_acceleration"] = True
-    cfg["save_to_xls"] = True
-    cfg["bin_num"] = 25
-    cfg["plot_SE"] = True
-    cfg["standardise_y_at_SC_level"] = False
-    cfg["standardise_y_to_a_joint"] = True
-    cfg["y_standardisation_joint"] = ["Knee"]  # "Hind paw tao"]
-    cfg["plot_joint_number"] = 3
-    cfg["color_palette"] = "viridis"
-    cfg["legend_outside"] = True
-    cfg["invert_y_axis"] = True
-    cfg["flip_gait_direction"] = True
-    cfg["analyse_average_x"] = True
-    cfg["standardise_x_coordinates"] = True
-    cfg["x_standardisation_joint"] = ["Hind paw tao"]
-    cfg["coordinate_standardisation_xls"] = ""
-    cfg["hind_joints"] = ["Hind paw tao", "Ankle", "Knee", "Hip", "Iliac Crest"]
-    cfg["fore_joints"] = [
-        "Front paw tao ",
-        "Wrist ",
-        "Elbow ",
-        "Lower Shoulder ",
-        "Upper Shoulder ",
-    ]
-    cfg["beam_col_left"] = ["BeamLeft"]  # BEAM_COL_LEFT & _RIGHT must be lists of len=1
-    cfg["beam_col_right"] = ["BeamRight"]
-    cfg["beam_hind_jointadd"] = ["Tail base ", "Tail center ", "Tail tip "]
-    cfg["beam_fore_jointadd"] = ["Nose ", "Ear base "]
-    cfg["angles"] = {
-        "name": ["Ankle ", "Knee ", "Hip "],
-        "lower_joint": ["Hind paw tao ", "Ankle ", "Knee "],
-        "upper_joint": ["Knee ", "Hip ", "Iliac Crest "],
-    }
-    return cfg
-
-
-# %%.......  main analysis: sc-lvl y-norm, features, df-creation & export ..........
-
-
-def test_height_standardisation_no_beam():
-    """Unit test normalising heights if there is no beam"""
-    # Create a sample DataFrame
-    data = {
-        "first_y": [-10, -5, 0, 10],
-        "second_y": [10, 20, 30, 40],
-        "third_y": [5, 15, 25, 35],
-    }
-    step = pd.DataFrame(data)
-    y_cols = [col for col in step.columns if col.endswith("y")]
-    this_y_min = step[y_cols].min().min()
-    step[y_cols] -= this_y_min
-    # Expected result
-    expected_result = {
-        "first_y": [0, 5, 10, 20],
-        "second_y": [20, 30, 40, 50],
-        "third_y": [15, 25, 35, 45],
-    }
-    expected_step = pd.DataFrame(expected_result)
-    # Compare the result with the expected result
-    pdt.assert_frame_equal(step, expected_step)
-
-
-cases = [
-    ((0, 0), (1, 0), (0, 1), 90), 
-    ((0, 0), (1, 0), (0, 0.5), 90),
-    ((0, 0), (1, 0), (2, 0), 0)
-]  # fmt: skip
-@pytest.mark.parametrize("angle_x_y, lower_x_y, upper_x_y, expected_angle", cases)
-def test_angles(angle_x_y, lower_x_y, upper_x_y, expected_angle):
-    step = (
-        pd.Series(
-            {
-                "angle x": angle_x_y[0],
-                "angle y": angle_x_y[1],
-                "lower x": lower_x_y[0],
-                "lower y": lower_x_y[1],
-                "upper x": upper_x_y[0],
-                "upper y": upper_x_y[1],
-            }
-        )
-        .to_frame()
-        .T
-    )
-    info = {}
-    cfg = {}
-    cfg["angles"] = {
-        "name": ["angle "],
-        "lower_joint": ["lower "],
-        "upper_joint": ["upper "],
-    }
-    step = add_angles(step, info, cfg)
-    assert step["angle Angle"].values == expected_angle
-
-
-def test_angles_not_depending_on_x_standardisation_and_gait_direction_flipping(
-    extract_info, extract_folderinfo, extract_cfg
-):
-    # 1. preparation
-    data = some_prep("DLC", extract_info, extract_folderinfo, extract_cfg)
-    all_cycles = extract_stepcycles(
-        "DLC", data, extract_info, extract_folderinfo, extract_cfg
-    )
-    # 2. x standardisation
-    results = analyse_and_export_stepcycles(data, all_cycles, extract_info, extract_cfg)
-    all_steps_data = results["all_steps_data"]
-    x_standardised_steps_data = results["x_standardised_steps_data"]
-    angle_cols = [col for col in all_steps_data.columns if col.endswith("Angle")]
-    for angle_col in angle_cols:
-        pdt.assert_series_equal(
-            all_steps_data[angle_col], x_standardised_steps_data[angle_col]
-        )
-    # 3. gait direction flipping
-    extract_cfg["flip_gait_direction"] = False
-    non_flipped_data = some_prep("DLC", extract_info, extract_folderinfo, extract_cfg)
-    non_flipped_results = analyse_and_export_stepcycles(
-        non_flipped_data, all_cycles, extract_info, extract_cfg
-    )
-    non_flipped_all_steps_data = non_flipped_results["all_steps_data"]
-    for angle_col in angle_cols:
-        pdt.assert_series_equal(
-            all_steps_data[angle_col], non_flipped_all_steps_data[angle_col]
-        )
-
-
-def test_velocities():
-    """Unit test of how velocities are added
-    A Note
-    ------
-    If you ever consider testing if velocities are expected after flipping x-values using flip_mouse_body - DON'T!
-    => We only flip if gait direction is right=>left in the first place and thus: velocities are CORRECT and comparable to non-flipped left=>right videos.
-    => NU - write a test that confirms this formally
-    """
-    data = {
-        "Sample x": [0.0, 2.0, 4.0, 8.0, 8.0, 4.0, 2.0, 0.0, -2.0, -10.0],
-        "Sample Angle": [-10.0, -8.0, -7.0, -4.0, 0.0, 4.0, 10.0, 20.0, 10.0, 0.0],
-    }
-    step = pd.DataFrame(data)
-    cfg = {
-        "hind_joints": ["Sample "],
-        "x_acceleration": True,
-        "angular_acceleration": True,
-    }
-    step = add_x_velocities(step, cfg)
-    step = add_angular_velocities(step, cfg)
-    # expected values were obtained by calling np.gradient on arrays above
-    expected_values = {
-        "Sample Velocity": [2.0, 2.0, 3.0, 2.0, -2.0, -3.0, -2.0, -2.0, -5.0, -8.0],
-        "Sample Acceleration": [0.0, 0.5, 0.0, -2.5, -2.5, 0.0, 0.5, -1.5, -3.0, -3.0],
-        "Sample Angle Velocity": [2.0, 1.5, 2.0, 3.5, 4.0, 5.0, 8.0, 0.0, -10.0, -10.0],
-        "Sample Angle Acceleration": [
-            -0.5,
-            0.0,
-            1.0,
-            1.0,
-            0.75,
-            2.0,
-            -2.5,
-            -9.0,
-            -5.0,
-            0.0,
-        ],
-    }
-    for key in expected_values.keys():
-        assert all(expected_values[key] == step[key])
-
-
-def test_step_separators():
-    """Unit test of how step separators are added to df"""
-    df_no_separators = pd.DataFrame(
-        {"A": [1, 2, 3, 4, 5], "B": ["yes", "no", "yes", "no", "yes"]}
-    )
-    nanvector = df_no_separators.loc[[1]]
-    nanvector[:] = np.nan
-    numvector = df_no_separators.loc[[1]]
-    numvector[:] = 1
-    df_manually_added_separators = pd.DataFrame(
-        {
-            "A": [1, 2, 3, 4, 5, np.nan, 1, np.nan],
-            "B": ["yes", "no", "yes", "no", "yes", np.nan, 1, np.nan],
-        }
-    )
-    df_manually_added_separators.index = [0, 1, 2, 3, 4, 1, 1, 1]
-    df_func_added_separators = add_step_separators(
-        df_no_separators, nanvector, numvector
-    )
-    pd.testing.assert_frame_equal(
-        df_func_added_separators, df_manually_added_separators
-    )