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