def test_storage_key_after_update(inplace): """ Assert that the storage key is not set after mutation of the index object """ original_index = ExplicitSecondaryIndex( column="col", index_dct={1: ["part_1", "part_2"], 3: ["part_3"]}, index_storage_key="storage_key", ) updated_index = original_index.remove_partitions([], inplace=inplace) assert updated_index.index_storage_key == "storage_key" updated_index = original_index.remove_partitions(["part_1"], inplace=inplace) assert updated_index.index_storage_key is None original_index = ExplicitSecondaryIndex( column="col", index_dct={1: ["part_1", "part_2"], 3: ["part_3"]}, index_storage_key="storage_key", ) updated_index = original_index.remove_values([], inplace=inplace) assert updated_index.index_storage_key == "storage_key" updated_index = original_index.remove_values([1], inplace=inplace) assert updated_index.index_storage_key is None original_index = ExplicitSecondaryIndex( column="col", index_dct={1: ["part_1", "part_2"], 3: ["part_3"]}, index_storage_key="storage_key", ) updated_index = original_index.copy() assert updated_index.index_storage_key == "storage_key" updated_index = original_index.copy(column="something_different") assert updated_index.index_storage_key is None
def test_index_remove_values(inplace): original_index = ExplicitSecondaryIndex( column="col", index_dct={1: ["part_1", "part_2"], 2: ["part_1"], 3: ["part_3"]} ) new_index = original_index.remove_values([1, 2], inplace=inplace) expected_index = ExplicitSecondaryIndex(column="col", index_dct={3: ["part_3"]}) assert new_index == expected_index