def test_identify_columns_that_repeat(self): columns_that_repeat1 = ntd.get_columns_that_appear_to_repeat(self.field_names, search_pattern="DX ") self.assertTrue(len(columns_that_repeat1)) first_element1 = columns_that_repeat1[0] self.assertEquals(tuple, first_element1.__class__) self.assertEquals(int, first_element1[1].__class__) self.assertEquals(1, first_element1[1]) columns_that_repeat2 = ntd.get_columns_that_appear_to_repeat(self.field_names, search_pattern="Proc_Code_") self.assertTrue(len(columns_that_repeat2)) first_element2 = columns_that_repeat2[0] self.assertEquals(tuple, first_element2.__class__) self.assertEquals(int, first_element2[1].__class__) self.assertEquals(1, first_element2[1])
def test_more_difficult_column_names_that_repeat(self): columns_that_repeat = ntd.get_columns_that_appear_to_repeat(self.more_difficult_field_name, "Other Procedure (ICD) Code ") self.assertTrue(len(columns_that_repeat))