94 lines
4.6 KiB
Python
94 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from datetime import datetime
|
|
from arttools import arttime
|
|
from astropy.time import Time
|
|
from arttools.arttime import TZ_UTC
|
|
from arttools.arttime import TZ_MSK
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
|
|
class TestConvertArtday(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self._test_conv = {'artday': 8000., 'missiontime': 691200000., 'mjd': 59543.875,
|
|
'datetime_msk': '2021-11-26T00:00:00', 'datetime_utc': '2021-11-25T21:00:00'}
|
|
self._test_mjd = Time(self._test_conv['mjd'], format='mjd')
|
|
self._test_dtime_utc = datetime.fromisoformat(self._test_conv['datetime_utc']).replace(tzinfo=TZ_UTC)
|
|
self._test_dtime_msk = datetime.fromisoformat(self._test_conv['datetime_msk']).replace(tzinfo=TZ_MSK)
|
|
|
|
def test_ArtTime_mjd(self):
|
|
mjd = arttime.ArtTime(mjd=self._test_conv['mjd'])
|
|
self.assertEqual(mjd.mjd.time, self._test_mjd)
|
|
|
|
def test_mjd_convert(self):
|
|
mjd = arttime.ArtTime(mjd=self._test_conv['mjd'])
|
|
test_dtime_utc = datetime.fromisoformat(self._test_conv['datetime_utc']).replace(tzinfo=TZ_UTC)
|
|
test_dtime_msk = datetime.fromisoformat(self._test_conv['datetime_msk']).replace(tzinfo=TZ_MSK)
|
|
self.assertEqual(mjd.artday.time, self._test_conv['artday'])
|
|
self.assertEqual(mjd.missiontime.time, self._test_conv['missiontime'])
|
|
self.assertEqual(mjd.datetime_utc.time, test_dtime_utc)
|
|
self.assertEqual(mjd.datetime_msk.time, test_dtime_msk)
|
|
|
|
def test_ArtTime_artday(self):
|
|
artday = arttime.ArtTime(artday=self._test_conv['artday'])
|
|
self.assertEqual(artday.artday.time, self._test_conv['artday'])
|
|
|
|
def test_artday_convert(self):
|
|
artday = arttime.ArtTime(artday=self._test_conv['artday'])
|
|
self.assertEqual(artday.mjd.time, self._test_mjd)
|
|
self.assertEqual(artday.missiontime.time, self._test_conv['missiontime'])
|
|
self.assertEqual(artday.datetime_utc.time, self._test_dtime_utc)
|
|
self.assertEqual(artday.datetime_msk.time, self._test_dtime_msk)
|
|
|
|
def test_ArtTime_missiontime(self):
|
|
missiontime = arttime.ArtTime(missiontime=self._test_conv['missiontime'])
|
|
self.assertEqual(missiontime.missiontime.time, self._test_conv['missiontime'])
|
|
|
|
def test_cyclic_conversion_missiontime(self):
|
|
missiontime = arttime.ArtTime(missiontime=self._test_conv['missiontime'])
|
|
datetime_utc = str(missiontime.datetime_utc)
|
|
test_datetime_utc = arttime.ArtTime(datetime_utc=datetime_utc)
|
|
self.assertEqual(missiontime.missiontime.time, test_datetime_utc.missiontime.time)
|
|
|
|
def test_missiontime_convert(self):
|
|
missiontime = arttime.ArtTime(missiontime=self._test_conv['missiontime'])
|
|
self.assertEqual(missiontime.mjd.time, self._test_mjd)
|
|
self.assertEqual(missiontime.artday.time, self._test_conv['artday'])
|
|
self.assertEqual(missiontime.datetime_utc.time, self._test_dtime_utc)
|
|
self.assertEqual(missiontime.datetime_msk.time, self._test_dtime_msk)
|
|
|
|
def test_ArtTime_datetime_utc(self):
|
|
datetime_utc = arttime.ArtTime(datetime_utc=self._test_conv['datetime_utc'])
|
|
self.assertEqual(datetime_utc.datetime_utc.time, self._test_dtime_utc)
|
|
|
|
def test_datetime_utc_convert(self):
|
|
datetime_utc = arttime.ArtTime(datetime_utc=self._test_conv['datetime_utc'])
|
|
self.assertEqual(datetime_utc.mjd.time, self._test_mjd)
|
|
self.assertEqual(datetime_utc.artday.time, self._test_conv['artday'])
|
|
self.assertEqual(datetime_utc.missiontime.time, self._test_conv['missiontime'])
|
|
self.assertEqual(datetime_utc.datetime_msk.time, self._test_dtime_msk)
|
|
|
|
def test_ArtTime_datetime_msk(self):
|
|
datetime_msk = arttime.ArtTime(datetime_msk=self._test_conv['datetime_msk'])
|
|
self.assertEqual(datetime_msk.datetime_msk.time, self._test_dtime_msk)
|
|
|
|
def test_datetime_msk_convert(self):
|
|
datetime_msk = arttime.ArtTime(datetime_msk=self._test_conv['datetime_msk'])
|
|
self.assertEqual(datetime_msk.mjd.time, self._test_mjd)
|
|
self.assertEqual(datetime_msk.artday.time, self._test_conv['artday'])
|
|
self.assertEqual(datetime_msk.missiontime.time, self._test_conv['missiontime'])
|
|
self.assertEqual(datetime_msk.datetime_utc.time, self._test_dtime_utc)
|
|
|
|
def test_ArtTime(self):
|
|
datetime_utcnow = arttime.ArtTime()
|
|
test_datetime_utcnow = datetime.utcnow().replace(tzinfo=TZ_UTC)
|
|
self.assertLess((test_datetime_utcnow - datetime_utcnow.datetime_utc.time).total_seconds(), 60.)
|
|
|
|
#### Add test for several parameters
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main() |