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()
