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.
180 Commits
10 Branches
26 MiB
 
 
 
 
Tree: 4ea148e8d1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '4ea148e8d1'
${ noResults }
kosmorro/test/dumper.py

328 lines
18 KiB
Raw Blame History 

  1. import unittest

  2. from datetime import date, datetime

  3. 

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

  5. from kosmorrolib.dumper import JsonDumper, TextDumper, _LatexDumper

  6. 

  7. 

  8. class DumperTestCase(unittest.TestCase):

  9.     def setUp(self) -> None:

  10.         self.maxDiff = None

  11. 

  12.     def test_json_dumper_returns_correct_json(self):

  13.         self.assertEqual('{\n'

  14.                          '    "ephemerides": [\n'

  15.                          '        {\n'

  16.                          '            "object": {\n'

  17.                          '                "name": "Mars",\n'

  18.                          '                "type": "planet",\n'

  19.                          '                "radius": null\n'

  20.                          '            },\n'

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

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

  23.                          '            "set_time": null\n'

  24.                          '        }\n'

  25.                          '    ],\n'

  26.                          '    "moon_phase": {\n'

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

  28.                          '        "time": "2019-10-14T00:00:00",\n'

  29.                          '        "next": {\n'

  30.                          '            "phase": "LAST_QUARTER",\n'

  31.                          '            "time": "2019-10-21T00:00:00"\n'

  32.                          '        }\n'

  33.                          '    },\n'

  34.                          '    "events": [\n'

  35.                          '        {\n'

  36.                          '            "objects": [\n'

  37.                          '                {\n'

  38.                          '                    "name": "Mars",\n'

  39.                          '                    "type": "planet",\n'

  40.                          '                    "radius": null\n'

  41.                          '                }\n'

  42.                          '            ],\n'

  43.                          '            "event": "OPPOSITION",\n'

  44.                          '            "starts_at": "2019-10-14T23:00:00",\n'

  45.                          '            "ends_at": null,\n'

  46.                          '            "details": null\n'

  47.                          '        },\n'

  48.                          '        {\n'

  49.                          '            "objects": [\n'

  50.                          '                {\n'

  51.                          '                    "name": "Venus",\n'

  52.                          '                    "type": "planet",\n'

  53.                          '                    "radius": null\n'

  54.                          '                }\n'

  55.                          '            ],\n'

  56.                          '            "event": "MAXIMAL_ELONGATION",\n'

  57.                          '            "starts_at": "2019-10-14T12:00:00",\n'

  58.                          '            "ends_at": null,\n'

  59.                          '            "details": "42.0\\u00b0"\n'

  60.                          '        }\n'

  61.                          '    ]\n'

  62.                          '}', JsonDumper(self._get_ephemerides(), self._get_moon_phase(), self._get_events()).to_string())

  63. 

  64.         self.assertEqual('{\n'

  65.                          '    "ephemerides": [\n'

  66.                          '        {\n'

  67.                          '            "object": {\n'

  68.                          '                "name": "Mars",\n'

  69.                          '                "type": "planet",\n'

  70.                          '                "radius": null\n'

  71.                          '            },\n'

  72.                          '            "rise_time": "2019-10-14T08:00:00",\n'

  73.                          '            "culmination_time": "2019-10-14T13:00:00",\n'

  74.                          '            "set_time": "2019-10-14T23:00:00"\n'

  75.                          '        }\n'

  76.                          '    ],\n'

  77.                          '    "moon_phase": {\n'

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

  79.                          '        "time": "2019-10-14T00:00:00",\n'

  80.                          '        "next": {\n'

  81.                          '            "phase": "LAST_QUARTER",\n'

  82.                          '            "time": "2019-10-21T00:00:00"\n'

  83.                          '        }\n'

  84.                          '    },\n'

  85.                          '    "events": [\n'

  86.                          '        {\n'

  87.                          '            "objects": [\n'

  88.                          '                {\n'

  89.                          '                    "name": "Mars",\n'

  90.                          '                    "type": "planet",\n'

  91.                          '                    "radius": null\n'

  92.                          '                }\n'

  93.                          '            ],\n'

  94.                          '            "event": "OPPOSITION",\n'

  95.                          '            "starts_at": "2019-10-14T23:00:00",\n'

  96.                          '            "ends_at": null,\n'

  97.                          '            "details": null\n'

  98.                          '        },\n'

  99.                          '        {\n'

  100.                          '            "objects": [\n'

  101.                          '                {\n'

  102.                          '                    "name": "Venus",\n'

  103.                          '                    "type": "planet",\n'

  104.                          '                    "radius": null\n'

  105.                          '                }\n'

  106.                          '            ],\n'

  107.                          '            "event": "MAXIMAL_ELONGATION",\n'

  108.                          '            "starts_at": "2019-10-14T12:00:00",\n'

  109.                          '            "ends_at": null,\n'

  110.                          '            "details": "42.0\\u00b0"\n'

  111.                          '        }\n'

  112.                          '    ]\n'

  113.                          '}', JsonDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  114.                                          self._get_events()).to_string())

  115. 

  116.     def test_text_dumper_without_events(self):

  117.         ephemerides = self._get_ephemerides()

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

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

  120.                          '--------  -----------  ------------------  ----------\n'

  121.                          'Mars           -               -               -\n\n'

  122.                          'Moon phase: Full Moon\n'

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

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

  125.                          TextDumper(ephemerides, self._get_moon_phase(), [], date=date(2019, 10, 14), with_colors=False).to_string())

  126. 

  127.         ephemerides = self._get_ephemerides(aster_rise_set=True)

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

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

  130.                          '--------  -----------  ------------------  ----------\n'

  131.                          'Mars         08:00           13:00           23:00\n\n'

  132.                          'Moon phase: Full Moon\n'

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

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

  135.                          TextDumper(ephemerides, self._get_moon_phase(), [], date=date(2019, 10, 14), with_colors=False).to_string())

  136. 

  137.     def test_text_dumper_with_events(self):

  138.         ephemerides = self._get_ephemerides()

  139.         self.assertEqual("Monday October 14, 2019\n\n"

  140.                          "Object     Rise time    Culmination time    Set time\n"

  141.                          "--------  -----------  ------------------  ----------\n"

  142.                          "Mars           -               -               -\n\n"

  143.                          "Moon phase: Full Moon\n"

  144.                          "Last Quarter on Monday October 21, 2019 at 00:00\n\n"

  145.                          "Expected events:\n"

  146.                          "23:00  Mars is in opposition\n"

  147.                          "12:00  Venus's largest elongation (42.0°)\n\n"

  148.                          "Note: All the hours are given in UTC.",

  149.                          TextDumper(ephemerides, self._get_moon_phase(), self._get_events(), date=date(2019, 10, 14), with_colors=False).to_string())

  150. 

  151.     def test_text_dumper_without_ephemerides_and_with_events(self):

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

  153.                          'Moon phase: Full Moon\n'

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

  155.                          'Expected events:\n'

  156.                          '23:00  Mars is in opposition\n'

  157.                          "12:00  Venus's largest elongation (42.0°)\n\n"

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

  159.                          TextDumper(None, self._get_moon_phase(), self._get_events(),

  160.                                     date=date(2019, 10, 14), with_colors=False).to_string())

  161. 

  162.     def test_timezone_is_taken_in_account(self):

  163.         ephemerides = self._get_ephemerides(aster_rise_set=True)

  164. 

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

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

  167.                          '--------  -----------  ------------------  -------------\n'

  168.                          'Mars         09:00           14:00         Oct 15, 00:00\n\n'

  169.                          'Moon phase: Full Moon\n'

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

  171.                          'Expected events:\n'

  172.                          'Oct 15, 00:00  Mars is in opposition\n'

  173.                          "13:00          Venus's largest elongation (42.0°)\n\n"

  174.                          'Note: All the hours are given in the UTC+1 timezone.',

  175.                          TextDumper(ephemerides, self._get_moon_phase(), self._get_events(), date=date(2019, 10, 14),

  176.                                     with_colors=False, timezone=1).to_string())

  177. 

  178.         ephemerides = self._get_ephemerides(aster_rise_set=True)

  179. 

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

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

  182.                          '--------  -----------  ------------------  ----------\n'

  183.                          'Mars         07:00           12:00           22:00\n\n'

  184.                          'Moon phase: Full Moon\n'

  185.                          'Last Quarter on Sunday October 20, 2019 at 23:00\n\n'

  186.                          'Expected events:\n'

  187.                          '22:00  Mars is in opposition\n'

  188.                          "11:00  Venus's largest elongation (42.0°)\n\n"

  189.                          'Note: All the hours are given in the UTC-1 timezone.',

  190.                          TextDumper(ephemerides, self._get_moon_phase(), self._get_events(), date=date(2019, 10, 14),

  191.                                     with_colors=False, timezone=-1).to_string())

  192. 

  193.     def test_latex_dumper(self):

  194.         latex = _LatexDumper(self._get_ephemerides(), self._get_moon_phase(), self._get_events(),

  195.                              date=date(2019, 10, 14)).to_string()

  196. 

  197.         self.assertRegex(latex, 'Monday October 14, 2019')

  198.         self.assertRegex(latex, 'Full Moon')

  199.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  200.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  201.         self.assertRegex(latex, r'\\object\{Mars\}\{-\}\{-\}\{-\}')

  202.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  203.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  204. 

  205.         latex = _LatexDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  206.                              self._get_events(), date=date(2019, 10, 14)).to_string()

  207.         self.assertRegex(latex, r'\\object\{Mars\}\{08:00\}\{13:00\}\{23:00\}')

  208. 

  209.     def test_latex_dumper_without_ephemerides(self):

  210.         latex = _LatexDumper(None, self._get_moon_phase(), self._get_events(),

  211.                              date=date(2019, 10, 14)).to_string()

  212. 

  213.         self.assertRegex(latex, 'Monday October 14, 2019')

  214.         self.assertRegex(latex, 'Full Moon')

  215.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  216.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  217.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  218. 

  219.         self.assertNotRegex(latex, r'\\object\{Mars\}\{-\}\{-\}\{-\}')

  220.         self.assertNotRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  221. 

  222.     def test_latex_dumper_without_events(self):

  223.         latex = _LatexDumper(self._get_ephemerides(), self._get_moon_phase(), [], date=date(2019, 10, 14)).to_string()

  224. 

  225.         self.assertRegex(latex, 'Monday October 14, 2019')

  226.         self.assertRegex(latex, 'Full Moon')

  227.         self.assertRegex(latex, r'\\object\{Mars\}\{-\}\{-\}\{-\}')

  228.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  229. 

  230.         self.assertNotRegex(latex, r'\\section{\\sffamily Expected events}')

  231. 

  232.     def test_latex_dumper_with_graph(self):

  233.         latex = _LatexDumper(self._get_ephemerides(True), self._get_moon_phase(), self._get_events(),

  234.                              date=date(2019, 10, 14), show_graph=True).to_string()

  235. 

  236.         self.assertRegex(latex, 'Monday October 14, 2019')

  237.         self.assertRegex(latex, 'Full Moon')

  238.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  239.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  240.         self.assertRegex(latex, r'\\graphobject\{18\}\{gray\}\{8\}\{0\}\{23\}\{0\}\{08:00\}\{23:00\}')

  241.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  242.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  243. 

  244.         latex = _LatexDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  245.                              self._get_events(), date=date(2019, 10, 14)).to_string()

  246.         self.assertRegex(latex, r'\\object\{Mars\}\{08:00\}\{13:00\}\{23:00\}')

  247. 

  248.     def test_latex_dumper_with_graph_but_without_rise_time(self):

  249.         ephemerides = self._get_ephemerides(True)

  250.         ephemerides[0].rise_time = None

  251.         latex = _LatexDumper(ephemerides, self._get_moon_phase(), self._get_events(),

  252.                              date=date(2019, 10, 14), show_graph=True).to_string()

  253. 

  254.         self.assertRegex(latex, 'Monday October 14, 2019')

  255.         self.assertRegex(latex, 'Full Moon')

  256.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  257.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  258.         self.assertRegex(latex, r'\\graphobject\{18\}\{gray\}\{0\}\{0\}\{23\}\{0\}\{\}\{23:00\}')

  259.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  260.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  261. 

  262.         latex = _LatexDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  263.                              self._get_events(), date=date(2019, 10, 14)).to_string()

  264.         self.assertRegex(latex, r'\\object\{Mars\}\{08:00\}\{13:00\}\{23:00\}')

  265. 

  266.     def test_latex_dumper_with_graph_but_without_set_time(self):

  267.         ephemerides = self._get_ephemerides(True)

  268.         ephemerides[0].set_time = None

  269.         latex = _LatexDumper(ephemerides, self._get_moon_phase(), self._get_events(),

  270.                              date=date(2019, 10, 14), show_graph=True).to_string()

  271. 

  272.         self.assertRegex(latex, 'Monday October 14, 2019')

  273.         self.assertRegex(latex, 'Full Moon')

  274.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  275.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  276.         self.assertRegex(latex, r'\\graphobject\{18\}\{gray\}\{8\}\{0\}\{24\}\{0\}\{08:00\}\{\}')

  277.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  278.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  279. 

  280.         latex = _LatexDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  281.                              self._get_events(), date=date(2019, 10, 14)).to_string()

  282.         self.assertRegex(latex, r'\\object\{Mars\}\{08:00\}\{13:00\}\{23:00\}')

  283. 

  284.     def test_latex_dumper_with_graph_but_mars_sets_tomorrow(self):

  285.         ephemerides = self._get_ephemerides(True)

  286.         ephemerides[0].set_time = datetime(2019, 10, 15, 1)

  287.         latex = _LatexDumper(ephemerides, self._get_moon_phase(), self._get_events(),

  288.                              date=date(2019, 10, 14), show_graph=True).to_string()

  289. 

  290.         self.assertRegex(latex, 'Monday October 14, 2019')

  291.         self.assertRegex(latex, 'Full Moon')

  292.         self.assertRegex(latex, r'\\section{\\sffamily Expected events}')

  293.         self.assertRegex(latex, r'\\section{\\sffamily Ephemerides of the day}')

  294.         self.assertRegex(latex, r'\\graphobject\{18\}\{gray\}\{8\}\{0\}\{24\}\{0\}\{08:00\}\{\}')

  295.         self.assertRegex(latex, r'\\graphobject\{18\}\{gray\}\{0\}\{0\}\{1\}\{0\}\{\}\{Oct 15, 01:00\}')

  296.         self.assertRegex(latex, r'\\event\{23:00\}\{Mars is in opposition\}')

  297.         self.assertRegex(latex, r"\\event\{12:00\}\{Venus's largest elongation \(42.0°\)\}")

  298. 

  299.         latex = _LatexDumper(self._get_ephemerides(aster_rise_set=True), self._get_moon_phase(),

  300.                              self._get_events(), date=date(2019, 10, 14)).to_string()

  301.         self.assertRegex(latex, r'\\object\{Mars\}\{08:00\}\{13:00\}\{23:00\}')

  302. 

  303.     @staticmethod

  304.     def _get_ephemerides(aster_rise_set=False) -> [AsterEphemerides]:

  305.         rise_time = datetime(2019, 10, 14, 8) if aster_rise_set else None

  306.         culmination_time = datetime(2019, 10, 14, 13) if aster_rise_set else None

  307.         set_time = datetime(2019, 10, 14, 23) if aster_rise_set else None

  308. 

  309.         return [AsterEphemerides(rise_time, culmination_time, set_time, Planet('Mars', 'MARS'))]

  310. 

  311.     @staticmethod

  312.     def _get_moon_phase():

  313.         return MoonPhase('FULL_MOON', datetime(2019, 10, 14), datetime(2019, 10, 21))

  314. 

  315.     @staticmethod

  316.     def _get_events():

  317.         return [Event('OPPOSITION',

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

  319.                       datetime(2019, 10, 14, 23, 00)),

  320.                 Event('MAXIMAL_ELONGATION',

  321.                       [Planet('Venus', 'VENUS')],

  322.                       datetime(2019, 10, 14, 12, 00), details='42.0°'),

  323.                 ]

  324. 

  325. 

  326. if __name__ == '__main__':

  327.     unittest.main()