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"華子")
