def test_number_of_books_by_genres(self):
Library(self.store, ["Fantasy", "Thriller"])
self.add_books([self.books[0], self.books[1], self.books[2]], Library)
actual_result = Library.number_of_books_by_genres()
self.assertEqual({'Fantasy': 2, 'Thriller': 1}, actual_result)
Library.add_book(self.books[3])
actual_result = Library.number_of_books_by_genres()
self.assertEqual({'Fantasy': 2, 'Thriller': 2}, actual_result)
Library.remove_book(self.books[1])
actual_result = Library.number_of_books_by_genres()
self.assertEqual({'Fantasy': 1, 'Thriller': 2}, actual_result)
Library.remove_book(self.books[0])
actual_result = Library.number_of_books_by_genres()
self.assertEqual({'Fantasy': 0, 'Thriller': 2}, actual_result)
os.remove(os.path.realpath(self.store))
def get_data(self):
"""
Gain access of the data about the books in the library.
"""
self.data = Library.number_of_books_by_genres()
self.genres = sorted(self.data.keys())
self.count = list(self.data[genre] for genre in self. genres)
