def check_for_empty_index(comparable: Compare):
"""
Find row number of empty values in the index column
"""
index_column = comparable.data_frame[idx.get_index_name(comparable)]
comparable.empty_index = [index for index, value in enumerate(index_column)
if value == Field.empty_string.value
or value != value or value is None]
def check_for_disjunctive_index(comparable_a: Compare, comparable_b: Compare):
index_name_a = idx.get_index_name(comparable_a)
index_name_b = idx.get_index_name(comparable_b)
comparable_a.disjunctive_index = comparable_a.data_frame[
~comparable_a.data_frame[index_name_a].isin(comparable_b.
data_frame[index_name_b])]
def drop_duplicate_index(comparable: Compare):
index_name = idx.get_index_name(comparable)
index_of_dup = comparable.duplicate_index[index_name].index.values
comparable.data_frame = comparable.data_frame.drop(index_of_dup)
def check_for_duplicate_index(comparable: Compare):
index_name = idx.get_index_name(comparable)
comparable.duplicate_index = comparable.data_frame[
comparable.data_frame.duplicated([index_name], keep=False)]
def test_get_index_name(self, mock_index):
comparable = Compare()
comparable.index_column_name = mock_index
assert index_validator.get_index_name(comparable) == 'id'
def drop_disjunctive_index(comparable: Compare):
index_name = idx.get_index_name(comparable)
index_of_disjunctive = comparable.disjunctive_index[
index_name].index.values
comparable.data_frame = comparable.data_frame.drop(index_of_disjunctive)