def test_load(self): """test load_data.py. Args: no """ load = LoadData('test_load_data') load.main() active = redis.StrictRedis(host="192.168.0.2", port=6379, db=6, password='******') # check up self.assertEqual(len(active.keys("*:*:2018-12-09 *")), 4, "total number check failed") self.assertEqual(len(active.keys("visit:*:2018-12-09 *")), 1, "visit incre data check failed") self.assertEqual(len(active.keys("search:*:2018-12-09 *")), 3, "search incre data check failed") self.assertEqual(active.zcard("visit:2018-12-09 00"), 1, "visit set num check failed") self.assertEqual(active.zcard("search:2018-12-09 01"), 3, "search set num check failed") self.assertEqual(active.get("stock:600796:jianpin"), "qjsh", "jianpin check failed") self.assertEqual(int(active.get("visit:000001:2018-12-09 00")), 33, " visit value check failed") index_visit_set = active.zrange("visit:2018-12-09 00", 0, -1).index("000001") self.assertEqual(active.zrange("visit:2018-12-09 00", index_visit_set, index_visit_set, withscores=True)[0][1], 33.0, "visit set value check failed") self.assertEqual(int(active.get("search:000007:2018-12-09 01")), 13, "search value check failed") self.assertEqual(int(active.get("search:000721:2018-12-09 01")), 13, "search value check failed") self.assertEqual(int(active.get("search:600796:2018-12-09 01")), 13, "search value check failed") search_set_keys = active.zrange("search:2018-12-09 01", 0, -1) index_search_set_1 = search_set_keys.index("000007") index_search_set_2 = search_set_keys.index("000721") index_search_set_3 = search_set_keys.index("600796") self.assertEqual(active.zrange("search:2018-12-09 01", index_search_set_1, index_search_set_1, withscores=True)[0][1], 13.0, "search set value check failed") self.assertEqual(active.zrange("search:2018-12-09 01", index_search_set_2, index_search_set_2, withscores=True)[0][1], 13.0, "search set value check failed") self.assertEqual(active.zrange("search:2018-12-09 01", index_search_set_3, index_search_set_3, withscores=True)[0][1], 13.0, "search set value check failed")
def _do(self, data): load_data = LoadData(data) load_data.main()