def test_customer_recommends(self): customer_dict = parse_customer_list("CodeIQ_customer_list_utf8.txt") sales_rank_list = parse_sales_ranking("CodeIQ_sales_ranking_utf8.txt") ans = {u"陽子": [u"サンフランシスコの老人", u"墨堤通り物語", u"徳島阿波踊り空港殺人事件"], u"次郎": [u"もうPythonしか愛せない", u"なんでもPythonプログラミング", u"Pythonを加速するためのC++"], u"太郎": [u"デフレと政治", u"これからのインド市場と日本", u"老後に備える外貨預金の真実"], u"華子": [u"サンフランシスコの老人", u"墨堤通り物語", u"徳島阿波踊り空港殺人事件"], u"美香": [u"サンフランシスコの老人", u"墨堤通り物語", u"徳島阿波踊り空港殺人事件"], u"おさむ": [u"もうPythonしか愛せない", u"なんでもPythonプログラミング", u"Pythonを加速するためのC++"]} for cus in customer_dict.values(): self.assertEqual(ans[cus.name], cus.get_recommends(sales_rank_list)[:3])
def test_parse_customer_list(self): customer_dict = parse_customer_list("CodeIQ_customer_list_utf8.txt") self.assertIsInstance(customer_dict["001"], Customer) self.assertIsInstance(customer_dict["002"], Customer) self.assertIsInstance(customer_dict["003"], Customer) self.assertIsInstance(customer_dict["004"], Customer) self.assertIsInstance(customer_dict["005"], Customer) self.assertIsInstance(customer_dict["006"], Customer) self.assertEqual(customer_dict["001"].name, u"太郎") self.assertEqual(customer_dict["002"].name, u"次郎") self.assertEqual(customer_dict["003"].name, u"陽子") self.assertEqual(customer_dict["004"].name, u"おさむ") self.assertEqual(customer_dict["005"].name, u"美香") self.assertEqual(customer_dict["006"].name, u"華子")