def test_timestream_write(self): ts = TimeStream() ts.version = 1 ts.create(self.tmp_path, ext="jpg") self.assertEqual(ts.path, self.tmp_path) self.assertDictEqual(ts.image_data, {}) self.assertDictEqual(ts.data, {}) for _ in range(10): img = TimeStreamImage() arr = np.arange(300, dtype="uint8") arr = arr.reshape((10, 10, 3)) date = dt.datetime.now() str_date = ts_format_date(date) img.pixels = arr img.datetime = date img.data["fake"] = True ts.write_image(img) self.assertIn(str_date, ts.image_data) self.assertDictEqual(img.data, ts.image_data[str_date]) self.assertTrue(path.exists, img.path)
def __exec__(self, context, *args): img = TimeStreamImage(args[0]) img.pixels = self.createCompoundImage(args[0]) return [img]