def test_index_update_wrong_col():
original_index = ExplicitSecondaryIndex(column="col", index_dct={1: ["part_1"]})
new_index = ExplicitSecondaryIndex(column="another_col", index_dct={1: ["part_4"]})
with pytest.raises(ValueError) as e:
original_index.update(new_index)
assert (
str(e.value)
== "Trying to update an index with the wrong column. Got `another_col` but expected `col`"
)
def test_index_update(inplace):
original_index = ExplicitSecondaryIndex(
column="col", index_dct={1: ["part_1", "part_2"], 3: ["part_3"]}
)
new_index = ExplicitSecondaryIndex(
column="col", index_dct={1: ["part_4"], 4: ["part_4"]}
)
updated_index = original_index.update(new_index, inplace=inplace)
expected_index = ExplicitSecondaryIndex(
column="col",
index_dct={1: ["part_2", "part_4", "part_1"], 3: ["part_3"], 4: ["part_4"]},
)
assert updated_index == expected_index
