예제 #1
0
파일: tank.py 프로젝트: cpcloud/span
    def __init__(self, path):
        super(TdtTankBase, self).__init__()

        tank_with_ext = path + os.extsep
        tev_path = tank_with_ext + self.raw_ext
        tsq_path = tank_with_ext + self.header_ext

        assert_nonzero_existing_file(tev_path)
        assert_nonzero_existing_file(tsq_path)

        self.path = path
        self.name = os.path.basename(path)
        self.age = _match_int(self.age_re, self.name)
        self.site = _match_int(self.site_re, self.name)
        istart = self.stsq.timestamp.index[0]
        iend = self.stsq.timestamp.index[-1]
        tstart = pd.datetime.fromtimestamp(self.stsq.timestamp[istart])
        self.__datetime = pd.Timestamp(tstart)
        self.time = self.__datetime.time()
        self.date = self.__datetime.date()
        self.fs = self.stsq.reset_index(drop=True).fs[0]
        self.start = self.__datetime
        tend = pd.datetime.fromtimestamp(self.stsq.timestamp[iend])
        self.end = pd.Timestamp(tend)
        self.duration = np.timedelta64(self.end - self.start)
예제 #2
0
파일: test_utils.py 프로젝트: cpcloud/span
    def test_assert_non_existing_file(self):
        name = self.name

        with open(name, 'wb') as tf:
            randn(2).tofile(tf)

        assert_nonzero_existing_file(name)

        os.remove(name)

        self.assertRaises(AssertionError, assert_nonzero_existing_file, name)