import unittest
from refill import ch_type_spec
import numpy as np

import logging
logging.basicConfig(filename='/var/www/html/data/log/tspeed.log', level=logging.DEBUG)
log = logging.getLogger('SpeedTestCase')

class SpeedTestCase(unittest.TestCase):
    def test_calculate(self):
        # commits = []
        # succs=[]
        # commits = [23,8,82,72,157,228,119,164,119,226,166,127,349,220,267,133,57,257,307,83,14,25,172,428,145,379,219,335,231,630]
        # succs=[0,0,0,0,0,0,0,0,1,1,0,0,0,0,2,0,0,0,0,1,1,0,0,1,0,2,0,1,0,0]
        # commits = [1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 2, 0, 0, 1, 0, 1, 1, 0, 0, 0, 1, 1, 0, 1, 0, 0, 1]
        # succs = [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]

        # commits = [41, 64, 42, 40, 35, 35, 26, 36, 17, 16, 0, 0, 0, 3, 13, 7, 0, 0, 2, 0, 0, 0, 4, 0, 1, 6, 0, 0, 0, 0]
        # succs   = [0,  2,  0,  0,  1,  0,  0,  0,  0,  1,  0, 0, 0, 0, 0,  0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
        commits = [67, 31, 0, 146, 207, 156, 157, 0, 41, 286, 333, 166, 76, 292, 32, 472, 53, 102, 0, 62, 115, 104, 107, 200, 190, 66, 94, 411, 259,
                   105]
        succs = [0, 0, 0, 2, 0, 1, 0, 0, 0, 0, 2, 2, 1, 1, 0, 2, 0, 0, 0, 0, 0, 0, 0, 3, 1, 0, 0, 4, 3, 0]

        pmf = ch_type_spec('bodian')
        pmf.calculate(np.array(commits),np.array(succs))


        pass

    def test_npsort(self):
        commits = [23, 8, 82, 72, 157, 228, 119, 164, 119, 226, 166, 127, 349, 220, 267, 133, 57, 257, 307, 83, 14, 25, 172, 428, 145, 379, 219, 335,
                   231, 630]
        succs = [0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 2, 0, 0, 0, 0, 1, 1, 0, 0, 1, 0, 2, 0, 1, 0, 0]
        datas = np.concatenate([[commits],[succs]],axis=0)
        # datas[0,:].sort()
        indexer = datas[0].argsort()
        view = datas[:,indexer]

        def unique(datas):
            succs = []
            commits = []

            last_commit = datas[0,0]
            last_succ = datas[1,0]
            for i in range(datas.shape[1]):
                commit = datas[0,i]
                succ = datas[1,i]
                if commit != last_commit:
                    succs.append(last_succ)
                    commits.append(last_commit)
                    last_commit = commit
                    last_succ = succ
                elif i == 0:
                    pass
                else:
                    last_succ += succ

            succs.append(last_succ)
            commits.append(last_commit)
            result = np.concatenate([[commits],[succs]],axis=0)
            return result
        
        unique(datas)

    def test_fsch(self):
        def is_fs(chname):
            if len(chname) < 2:
                return False;
            ext = chname[-2:]
            return True if ext == 'fs' else False

        x = 'yongchonggongfs'
        k = is_fs(x)
        
        x = int(round(2.55,0))
        pass

    def testTime(self):
        import time
        a1 = "2019-5-10 23:40:00"
        # 先转换为时间数组
        timeArray = time.strptime(a1, "%Y-%m-%d %H:%M:%S")

        # 转换为时间戳
        timeStamp = int(time.mktime(timeArray))
        print(timeStamp)


if __name__ == '__main__':
    unittest.main()