def test_blank_skip_columns(self): column_names = ['A', 'B', 'C', 'D', 'E'] skip_cols = [] self.assertEqual(column_names, list(helpers.column_range(0, 5, skip_cols)))
def test_return_multiple(self): self.assertEqual(['A', 'B'], list(helpers.column_range(0, 2)))
def test_skip_columns(self): skip_cols = ['A', 'B', 'C', 'D'] self.assertEqual(['E'], list(helpers.column_range(0, 5, skip_cols)))
def test_return_A(self): self.assertEqual(['A'], list(helpers.column_range(0, 1)))
def test_blank_skip_columns(self): column_names = ['A', 'B', 'C', 'D', 'E'] skip_columns = [] self.assertEqual(column_names, list(helpers.column_range(0, 5, \ skip_columns)))
def test_skip_columns(self): skip_columns = ['A', 'B', 'C', 'D'] self.assertEqual(['E'], list(helpers.column_range(0, 5, skip_columns)))