Skip to content

Commit 54c03fe

Browse files
authored
add test for select except after gen / merge (#1407)
1 parent 586abbf commit 54c03fe

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/unit/lib/test_datachain.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1072,6 +1072,19 @@ def test_select_except(test_session):
10721072
assert n == len(features_nested)
10731073

10741074

1075+
def test_select_except_after_gen(test_session):
1076+
# https://github.com/iterative/datachain/issues/1359
1077+
# fixed by https://github.com/iterative/datachain/pull/1400
1078+
chain = dc.read_values(id=range(10), session=test_session)
1079+
1080+
chain = chain.gen(lambda id: [(id, 0)], output={"id": int, "x": int})
1081+
chain = chain.select_except("x")
1082+
chain = chain.merge(chain, on="id")
1083+
chain = chain.select_except("right_id")
1084+
1085+
assert set(chain.to_values("id")) == set(range(10))
1086+
1087+
10751088
def test_select_wrong_type(test_session):
10761089
chain = dc.read_values(fr1=features_nested, fr2=features, session=test_session)
10771090

0 commit comments

Comments
 (0)