lib/tests/data.py

import unittest

from kosmorrolib import model, core
from kosmorrolib.enum import ObjectIdentifier


class DataTestCase(unittest.TestCase):
    def test_object_radius_must_be_set_to_get_apparent_radius(self):
        o = model.Planet(ObjectIdentifier.SATURN, "SATURN")

        with self.assertRaises(ValueError) as context:
            o.get_apparent_radius(
                core.get_timescale().now(), core.get_skf_objects()["earth"]
            )

        self.assertEqual(("Missing radius for SATURN",), context.exception.args)


if __name__ == "__main__":
    unittest.main()