def test_parse_date_valid(self):
"""Test timestream.parse.ts_parse_date with a valid date"""
date_str = "2013_12_11_10_09_08"
date_obj = dt.datetime(2013, 12, 11, 10, 9, 8)
self.assertEqual(ts_parse_date(date_str), date_obj)
# ensure we can pass a datetime and have it silently returned.
self.assertEqual(ts_parse_date(date_obj), date_obj)
def test_get_image_good_datetime(self):
"""Test ts_get_image with a datetime obj on a good timestream"""
for iii in range(len(helpers.TS_MANIFOLD_DATES)):
date = ts_parse_date(helpers.TS_MANIFOLD_DATES[iii])
ts = helpers.FILES["timestream_manifold"]
res = ts_get_image(ts, date)
self.assertEqual(res, helpers.TS_MANIFOLD_FILES_JPG[iii])
def readTimeStampDataCSV(CSVFile, StartDate, EndDate):
TimestampStart = ts_parse_date(StartDate)
TimestampeEnd = ts_parse_date(EndDate)
with open(CSVFile, 'rb') as csvfile:
CsvReader = csv.reader(csvfile, delimiter=',')
Data = []
DateList = []
PotIndex = []
for i, Row in enumerate(CsvReader):
if i == 0:
PotIndex = [int(Num) for Num in Row[1:]]
else:
Timestamp = ts_parse_date(Row[0][:19])
if Timestamp.date() >= TimestampStart.date() and \
Timestamp.date() <= TimestampeEnd.date():
if Row[-1] == "Na":
Row[-1] = "NaN"
DataRow = [float(Num) for Num in Row[1:]]
Data.append(DataRow)
DateList.append(Timestamp)
return Data, DateList, PotIndex
def __init__(self, dt=None):
"""Initialise a TimeStreamImage
:param dt: The timestamp for this image.
:type dt: datetime.datetime object
"""
self._datetime = None
if dt:
self._datetime = ts_parse_date(dt)
self._timestream = None
self._path = None
self._pixels = None
self._ipm = None
self.data = {}
def test_parse_date_invalid(self):
"""Test timestream.parse.ts_parse_date with a valid date"""
# Bad format
date_str = "2013-12-11-10-09-08"
with self.assertRaises(ValueError):
ts_parse_date(date_str)
# Bad date
date_str = "2013_14_11_10_09_08"
with self.assertRaises(ValueError):
ts_parse_date(date_str)
# Missing time
date_str = "2013_12_11"
with self.assertRaises(ValueError):
ts_parse_date(date_str)
def datetime(self, dte):
self._datetime = ts_parse_date(dte)
self._path = None
def test_get_image_missing_datetime(self):
"""Test ts_get_image with a missing datetime on a good timestream"""
date = ts_parse_date("2010_10_10_10_10_10")
ts = helpers.FILES["timestream_manifold"]
res = ts_get_image(ts, date)
self.assertEqual(res, None)
"""
opts = docopt.docopt(CLI_OPTS)
interval = 24*60*60 #None
start = None
end = None
start_hour = None
end_hour = None
delay = 500
inputRootPath = opts['-i']
if opts['-d']:
delay = int(opts['-d'])
if opts['--int']:
interval = int(opts['-int'])
if opts['-s']:
start = ts_parse_date(opts['-s'])
if opts['-e']:
end = ts_parse_date(opts['-e'])
if opts['--sh']:
start_hour = datetime.time(int(opts['--sh']), 0, 0)
if opts['--eh']:
end_end = datetime.time(int(opts['--eh']), 0, 0)
if opts['-o']:
outputRootPath = opts['-o']
else:
outputRootPath = None
timestream.setup_module_logging(level=logging.INFO)
ts = timestream.TimeStream()
ts.load(inputRootPath)
