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)))
def test_return_multiple(self):
self.assertEqual(['A', 'B'], list(helpers.column_range(0, 2)))
def test_return_A(self):
self.assertEqual(['A'], list(helpers.column_range(0, 1)))
