Kosmorro
/
kosmorro
mirror of https://github.com/Kosmorro/kosmorro
1
0
Fork 0
Code Issues Releases 33 Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 Commits
10 Branches
13 MiB
 
 
 
 
Tree: 058254e702
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '058254e702'
${ noResults }
kosmorro/test/dumper.py

96 lines
4.8 KiB
Raw Blame History 

  1. import unittest

  2. from datetime import date

  3. 

  4. from kosmorrolib.data import AsterEphemerides, Planet, MoonPhase, Event

  5. from kosmorrolib.dumper import JsonDumper, TextDumper

  6. from kosmorrolib.core import get_timescale

  7. 

  8. 

  9. class DumperTestCase(unittest.TestCase):

  10.     def setUp(self) -> None:

  11.         self.maxDiff = None

  12. 

  13.     def test_json_dumper_returns_correct_json(self):

  14.         data = self._get_data()

  15.         self.assertEqual('{\n'

  16.                          '    "moon_phase": {\n'

  17.                          '        "next_phase_date": "2019-10-21T00:00:00Z",\n'

  18.                          '        "phase": "FULL_MOON",\n'

  19.                          '        "date": "2019-10-14T00:00:00Z"\n'

  20.                          '    },\n'

  21.                          '    "events": [\n'

  22.                          '        {\n'

  23.                          '            "event_type": "OPPOSITION",\n'

  24.                          '            "objects": [\n'

  25.                          '                "Mars"\n'

  26.                          '            ],\n'

  27.                          '            "start_time": "2018-07-27T05:12:00Z",\n'

  28.                          '            "end_time": null\n'

  29.                          '        }\n'

  30.                          '    ],\n'

  31.                          '    "ephemerides": [\n'

  32.                          '        {\n'

  33.                          '            "object": "Mars",\n'

  34.                          '            "details": {\n'

  35.                          '                "rise_time": null,\n'

  36.                          '                "culmination_time": null,\n'

  37.                          '                "set_time": null\n'

  38.                          '            }\n'

  39.                          '        }\n'

  40.                          '    ]\n'

  41.                          '}', JsonDumper(data,

  42.                                          [Event('OPPOSITION', [Planet('Mars', 'MARS')],

  43.                                                 get_timescale().utc(2018, 7, 27, 5, 12))]

  44.                                          ).to_string())

  45. 

  46.     def test_text_dumper_without_events(self):

  47.         ephemerides = self._get_data()

  48.         self.assertEqual('Monday October 14, 2019\n\n'

  49.                          'Object     Rise time    Culmination time    Set time\n'

  50.                          '--------  -----------  ------------------  ----------\n'

  51.                          'Mars           -               -               -\n\n'

  52.                          'Moon phase: Full Moon\n'

  53.                          'Last Quarter on Monday October 21, 2019 at 00:00\n\n'

  54.                          'Note: All the hours are given in UTC.',

  55.                          TextDumper(ephemerides, [], date=date(2019, 10, 14)).to_string())

  56. 

  57.     def test_text_dumper_with_events(self):

  58.         ephemerides = self._get_data()

  59.         self.assertEqual('Monday October 14, 2019\n\n'

  60.                          'Object     Rise time    Culmination time    Set time\n'

  61.                          '--------  -----------  ------------------  ----------\n'

  62.                          'Mars           -               -               -\n\n'

  63.                          'Moon phase: Full Moon\n'

  64.                          'Last Quarter on Monday October 21, 2019 at 00:00\n\n'

  65.                          'Expected events:\n\n'

  66.                          '05:12  Mars is in opposition\n\n'

  67.                          'Note: All the hours are given in UTC.',

  68.                          TextDumper(ephemerides, [Event('OPPOSITION',

  69.                                                         [Planet('Mars', 'MARS')],

  70.                                                         get_timescale().utc(2018, 7, 27, 5, 12))

  71.                                                   ], date=date(2019, 10, 14)).to_string())

  72. 

  73.     def test_text_dumper_without_ephemerides_and_with_events(self):

  74.         ephemerides = self._get_data(False)

  75.         self.assertEqual('Monday October 14, 2019\n\n'

  76.                          'Moon phase: Full Moon\n'

  77.                          'Last Quarter on Monday October 21, 2019 at 00:00\n\n'

  78.                          'Expected events:\n\n'

  79.                          '05:12  Mars is in opposition\n\n'

  80.                          'Note: All the hours are given in UTC.',

  81.                          TextDumper(ephemerides, [Event('OPPOSITION',

  82.                                                         [Planet('Mars', 'MARS')],

  83.                                                         get_timescale().utc(2018, 7, 27, 5, 12))

  84.                                                   ], date=date(2019, 10, 14)).to_string())

  85. 

  86.     @staticmethod

  87.     def _get_data(has_ephemerides: bool = True):

  88.         return {

  89.             'moon_phase': MoonPhase('FULL_MOON', get_timescale().utc(2019, 10, 14), get_timescale().utc(2019, 10, 21)),

  90.             'details': [Planet('Mars', 'MARS', AsterEphemerides(None, None, None))] if has_ephemerides else []

  91.         }

  92. 

  93. 

  94. if __name__ == '__main__':

  95.     unittest.main()