testPlot.py 7.7 KB

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