def test_get_header_names_with_no_names_and_no_header(self): args = MagicMock(names=None, input_options=['noheader']) header, names = io_lib.get_header_names(args) self.assertEqual(header, None) self.assertEqual(names, None)
def test_get_header_names_with_names_and_header(self): args = MagicMock(names=['a'], input_options=[]) header, names = io_lib.get_header_names(args) self.assertEqual(header, 0) self.assertEqual(names, ['a'])