def setUp(self): scrapy.log.start() self.validator = JmItemValidator() self.validator.brand_english = {} # english name keys are lower case self.validator.brand_english["adidas"] = 215 self.validator.brand_english["laneige"] = 21 self.validator.brand_english["clear"] = 239 self.validator.brand_english["versace"] = 248 self.validator.brand_english["yuejimei"] = 362 self.validator.brand_english["fanxi"] = 415 self.validator.brand_english["estee lauder"] = 505 self.validator.brand_english["burt's_bees"] = 255 self.validator.brand_english["2n"] = 488 self.validator.brand_english["it’s skin"] = 999 self.validator.brand_english["unknown"] = 0 self.validator.brand_chinese = {} self.validator.brand_chinese["阿迪达斯"] = 215 self.validator.brand_chinese["兰芝"] = 21 self.validator.brand_chinese["范思哲"] = 248 self.validator.brand_chinese["清扬"] = 239 self.validator.brand_chinese["悦己美"] = 362 self.validator.brand_chinese["凡茜"] = 415 self.validator.brand_chinese["雅诗兰黛"] = 505 self.validator.brand_chinese["小蜜蜂"] = 255 self.validator.brand_chinese["2n"] = 488 self.validator.brand_chinese["假肌肤"] = 999 self.validator.brand_chinese["unknown"] = 0
class TestJMPipelines(unittest.TestCase): def setUp(self): scrapy.log.start() self.validator = JmItemValidator() self.validator.brand_english = {} # english name keys are lower case self.validator.brand_english["adidas"] = 215 self.validator.brand_english["laneige"] = 21 self.validator.brand_english["clear"] = 239 self.validator.brand_english["versace"] = 248 self.validator.brand_english["yuejimei"] = 362 self.validator.brand_english["fanxi"] = 415 self.validator.brand_english["estee lauder"] = 505 self.validator.brand_english["burt's_bees"] = 255 self.validator.brand_english["2n"] = 488 self.validator.brand_english["it’s skin"] = 999 self.validator.brand_english["unknown"] = 0 self.validator.brand_chinese = {} self.validator.brand_chinese["阿迪达斯"] = 215 self.validator.brand_chinese["兰芝"] = 21 self.validator.brand_chinese["范思哲"] = 248 self.validator.brand_chinese["清扬"] = 239 self.validator.brand_chinese["悦己美"] = 362 self.validator.brand_chinese["凡茜"] = 415 self.validator.brand_chinese["雅诗兰黛"] = 505 self.validator.brand_chinese["小蜜蜂"] = 255 self.validator.brand_chinese["2n"] = 488 self.validator.brand_chinese["假肌肤"] = 999 self.validator.brand_chinese["unknown"] = 0 def test_get_brandid(self): item = JmProductItem() item['brand'] = "不知道" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 0) item['brand'] = "凡茜" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 415) item['brand'] = "悦己美(Please Me)" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 362) item['brand'] = "雅诗兰黛 (Estee Lauder)" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 505) item['brand'] = "2N" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 488) item['brand'] = "范思哲 versace" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 248) item['brand'] = "小蜜蜂burt's_bees" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 255) item['brand'] = "真肌肤 It’s skin" print "test get brand id, brand = ", item['brand'] bid = self.validator.get_brand_id(item) self.assertEqual(bid, 999)