def task_2(uuid): """Perform task from task 3 :param uuid: """ bs = DocumentAnalysis(dt) try: # By Countries result_countries = bs.countries_by_book(uuid) print("Views by country and Document UUID") print(result_countries) bs.counries_by_book_plot(result_countries, uuid) # By Continent print("Views by continent and Document UUID") result_continent = bs.continent_by_book(result_countries, uuid) bs.continent_by_book_plot(result_continent, uuid) print(result_continent) except Exception as err: print((str(err)))
def test_task_2_2(self): bs = DocumentAnalysis(self.dt) """ Test if expetion is raised for bad uuid""" result_countries = bs.countries_by_book(self.book_uuid) self.assertRaises(Exception, bs.continent_by_book(result_countries, "110727005030-00"))
def test_task_2_1(self): """Check if outut is correct""" bs = DocumentAnalysis(self.dt) result_countries = bs.countries_by_book(self.book_uuid) result_continent = bs.continent_by_book(result_countries, self.book_uuid) self.assertEqual(result_continent, {'EU': 2, 'NA': 1, 'SA': 23, 'OC': 1})