def test_process_books(self):
converters = [pipe_to_book, slash_to_book, csv_to_book]
file_parse_info = zip(get_input_files(), ["|", "/", ","], converters)
books = get_books_from_files(file_parse_info)
self.assertEquals(len(process_books(books, "", False, False)), 9)
books = get_books_from_files(file_parse_info)
self.assertEquals(process_books(books, None, False, False)[0],
Book("Kent", "Beck", "Implementation Patterns",
"2007"))
books = get_books_from_files(file_parse_info)
self.assertEquals(process_books(books, None, False, True)[-1],
Book("Kent", "Beck", "Implementation Patterns",
"2007"))
books = get_books_from_files(file_parse_info)
self.assertEquals(process_books(books, None, True, False)[0],
Book("Fred", "Brooks", "The Mythical Man-Month",
"1975"))
def test_row_from_file(self):
parse_info = zip(get_input_files(), [4, 3, 2], ["|", "/", ","])
for (file_path, nb_lines, delimiter) in parse_info:
self.assertEquals(len(list(rows_from_file(file_path, delimiter))),
nb_lines)
def test_get_books_from_files(self):
parse_info = zip(get_input_files(), ["|", "/", ","],
[pipe_to_book, slash_to_book, csv_to_book])
self.assertEquals(len(list(get_books_from_files(parse_info))), 9)
def test_get_input_files(self):
for file_path in get_input_files():
self.assertTrue(os.path.isfile(file_path))
