testPlot.py 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253
  1. import time
  2. import unittest
  3. import logging
  4. from refill import time_border
  5. logging.basicConfig(filename='/var/www/html/data/log/qreader.log', level=logging.DEBUG)
  6. log = logging.getLogger('reader')
  7. class MyTestCase(unittest.TestCase):
  8. # __redis_host = '192.168.3.104'
  9. __redis_host = '192.168.3.46'
  10. def test_something(self):
  11. self.assertEqual(True, False) # add assertion here
  12. def test_listener(self):
  13. try:
  14. from refill import queueListener
  15. queueListener.set_redis(self.__redis_host, '6379')
  16. queueListener.prepare_data()
  17. except Exception as ex:
  18. log.error(ex)
  19. pass
  20. def test_timestart(self):
  21. stamp = int(time.time())
  22. for i in range(300):
  23. x = stamp + i
  24. l = time_border(300, x, True)
  25. r = time_border(300, x, False)
  26. print('l=', l, 'r=', r)
  27. def testChannel(self):
  28. from refill import ChannelReader
  29. reader = ChannelReader()
  30. days = reader.days()
  31. print('days=', days)
  32. for day in days:
  33. x = reader.tuple_path(day, card_types={4, 5, 6})
  34. x1 = reader.tuple_path(day, {'chizeng', 'ainika'}, {4, 5, 6})
  35. x2 = reader.tuple_path(day, {'chizeng', 'ainika'}, {4, 5, 6}, 50)
  36. def test_chpainter(self):
  37. from refill import ChannelCumPainter
  38. start_time = int(time.time()) - 10 * 86400 - 3600
  39. end_time = int(time.time()) - 8 * 86400
  40. painter = ChannelCumPainter(start_time=start_time, end_time=end_time, chnames=set(), card_types={4, 5, 6})
  41. painter.paint()
  42. def test_chcov_ratio(self):
  43. from refill import ChannelCovPainter
  44. start_time = int(time.time()) - 10 * 86400 - 3600
  45. end_time = int(time.time()) - 10 * 86400
  46. painter = ChannelCovPainter(start_time=start_time, end_time=end_time, chnames=set(), card_types={4, 5, 6},filter_wave=3600)
  47. painter.paint()
  48. def test_chcov_succ(self):
  49. from refill import ChannelCovSuccPainter
  50. start_time = int(time.time()) - 20 * 86400 - 3600
  51. end_time = int(time.time()) - 20 * 86400
  52. painter = ChannelCovSuccPainter(start_time=start_time, end_time=end_time, chnames=set(), card_types={4, 5, 6},filter_wave=3600)
  53. painter.paint()
  54. def test_mch_ratio_painter(self):
  55. from refill import MerchantCumRatioPainter
  56. start_time = int(time.time()) - 21 * 86400 - 3600
  57. end_time = int(time.time()) - 20 * 86400
  58. painter = MerchantCumRatioPainter(start_time=start_time, end_time=end_time, mchids=set(), card_types={4, 5, 6})
  59. painter.paint()
  60. def test_mch_covratio_painter(self):
  61. from refill import MerchantCovRatioPainter
  62. start_time = int(time.time()) - 21 * 86400 - 3600
  63. end_time = int(time.time()) - 20 * 86400
  64. painter = MerchantCovRatioPainter(start_time=start_time, end_time=end_time, mchids=set(), card_types={4, 5, 6})
  65. painter.paint()
  66. def test_mch_amount_painter(self):
  67. from refill import MerchantAmountPainter
  68. start_time = int(time.time()) - 20 * 86400 - 3600
  69. end_time = int(time.time()) - 20 * 86400
  70. painter = MerchantAmountPainter(start_time=start_time, end_time=end_time, mchids=set(), card_types={4, 5, 6})
  71. painter.paint()
  72. def test_netcheck(self):
  73. from refill import NetchkReader
  74. import time
  75. from refill import day_stamp
  76. day = day_stamp(int(time.time()) - 4 * 86400)
  77. days = [day,day-86400]
  78. net = NetchkReader()
  79. channels = net.tuple_path(day)
  80. channels = net.many_tuple_path(days)
  81. def test_net_painter(self):
  82. from refill import NetcheckCovPainter
  83. start_time = int(time.time()) - 5 * 86400
  84. end_time = int(time.time()) - 4 * 86400
  85. painter = NetcheckCovPainter(start_time=start_time, end_time=end_time)
  86. painter.paint()
  87. def testDays(self):
  88. from refill import MerchantReader
  89. from refill import ChannelReader
  90. try:
  91. chreader = ChannelReader()
  92. chdays = chreader.days()
  93. xlables = [time.strftime('%d-%H:%M:%S', time.localtime(d)) for d in chdays]
  94. print(xlables)
  95. reader = MerchantReader()
  96. days = reader.days()
  97. except Exception as ex:
  98. log.error(ex)
  99. pass
  100. def test_jsonLoads(self):
  101. import json
  102. try:
  103. str = 4
  104. x = json.loads(str)
  105. print(x)
  106. except Exception as ex:
  107. print(ex)
  108. def test_partial(self):
  109. from functools import partial
  110. add_five = partial()
  111. def test_key(self):
  112. def person(name, age, *, city, job):
  113. print(name, age, city, job)
  114. person('Jack', 24, city='Beijing', job='Engineer')
  115. person('Jack', 24, 'Beijing', job='Engineer')
  116. person('Jack', 24, 'Beijing', 'Engineer')
  117. def test_none(self):
  118. x = None
  119. len = len(x)
  120. print(len)
  121. def test_set(self):
  122. x = set([(4, 50), (4, 100)])
  123. y = set([(4, 50), (5, 100)])
  124. z = x | y
  125. print(z)
  126. def test_env(self):
  127. import os
  128. x = os.getenv("PYCHARM_DISPLAY_PORT", "-1")
  129. print(x)
  130. def test_matplot(self):
  131. import matplotlib
  132. x = matplotlib.__version__
  133. print(x)
  134. def test_time(self):
  135. x = int(time.time())
  136. print(x)
  137. def test_rpop(self):
  138. import redis
  139. import json
  140. pool = redis.ConnectionPool(host=self.__redis_host, port=6379, db=0)
  141. r = redis.Redis(connection_pool=pool)
  142. item = r.rpop('REFILL_MONITOR_QUEUE')
  143. if item is None:
  144. print('hello')
  145. else:
  146. try:
  147. val = json.loads(item)
  148. method = val['method']
  149. params = val['params']
  150. print(method, params)
  151. except Exception as ex:
  152. log.error(ex)
  153. def test_consumer(self):
  154. from refill import WriterConsumer
  155. class PrintHandler:
  156. def write(self, method, msg):
  157. log.debug(msg)
  158. time.sleep(0.01)
  159. handler = PrintHandler()
  160. consumer = WriterConsumer(handler,'PrintHandler')
  161. consumer.start()
  162. for i in range(100000):
  163. consumer.put('test', f'index = {i}')
  164. consumer.quit()
  165. consumer.join()
  166. def test_merge(self):
  167. from collections import defaultdict
  168. import time as time
  169. def merge(l,r):
  170. for name,ls in l.items():
  171. if name in r:
  172. ls.extend(r[name])
  173. k = set(ls)
  174. r[name] = list(k)
  175. else:
  176. r[name] = ls
  177. return r
  178. all = defaultdict(list)
  179. a = {'yunchonggongfs': [(4, 100), (4, 30), (4, 50)]}
  180. b = {'yunchonggongfs': [(4, 100), (4, 200), (4, 50)]}
  181. all = merge(a,all)
  182. all = merge(b,all)
  183. x = 0;
  184. def test_split(self):
  185. def split_card(card_specs):
  186. result = dict()
  187. for card_type,spec in card_specs:
  188. if card_type not in result:
  189. result[card_type] = []
  190. result[card_type].append(spec)
  191. return result
  192. tups = [(4, 100), (4, 50) ,(5,10),(4, 30),(5,30),(6,100),(5,100),]
  193. x = split_card(tups)
  194. y = 1
  195. if __name__ == '__main__':
  196. unittest.main()