From 90e6915b5ec7f3e6ff00775ffd24c03583c3e8ae Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Tue, 9 Jul 2024 13:24:11 -0700 Subject: [PATCH] make first value b --- tests/nested_pandas/utils/test_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/nested_pandas/utils/test_utils.py b/tests/nested_pandas/utils/test_utils.py index 009a9c2..5397d75 100644 --- a/tests/nested_pandas/utils/test_utils.py +++ b/tests/nested_pandas/utils/test_utils.py @@ -15,7 +15,7 @@ def test_count_nested(join): data={ "c": [0, 2, 4, 1, np.nan, 3, 1, 4, 1], "d": [5, 4, 7, 5, 3, 1, 9, 3, 4], - "label": ["a", "a", "b", "b", "a", "a", "b", "a", "b"], + "label": ["b", "a", "b", "b", "a", "a", "b", "a", "b"], }, index=[0, 0, 0, 1, 1, 1, 2, 2, 2], ) @@ -28,8 +28,8 @@ def test_count_nested(join): # Test count by label_counts = count_nested(base, "nested", by="label", join=join) - assert all(label_counts["n_nested_a"].values == [2, 2, 1]) - assert all(label_counts["n_nested_b"].values == [1, 1, 2]) + assert all(label_counts["n_nested_a"].values == [1, 2, 1]) + assert all(label_counts["n_nested_b"].values == [2, 1, 2]) # Make sure the ordering is alphabetical # https://github.com/lincc-frameworks/nested-pandas/issues/109