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