Upload New File

tests/test_main.py

+import unittest
+from CpicImgSim import observation_simulation, quick_run
+
+
+class TestMain(unittest.TestCase):
+    def test_main(self):
+        target_example = {
+            'name': 'test',
+            'cstar': {
+                'magnitude': 0,
+                'ra': '120d',
+                'dec': '40d',
+                'distance': 10,
+                'sptype': 'F0III',
+            },
+            # 'stars': [
+            #     {
+            #         'magnitude': 15,
+            #         'ra': '120d0m5s',
+            #         'dec': '40d00m3s',
+            #         'sptype': 'G0III',
+            #         'isblackbody': True
+            #     },
+            # ],
+            # 'planets': [
+            #     {
+            #         'radius': 20,
+            #         'pangle': 60,
+            #         'separation': 0.7,
+            #         'phase_angle': 90,
+            #     }
+            # ]
+        }
+
+        quick_run('', 10, 'f661', 1, 1, 30)
+        quick_run('*2.4/10(3,5)/15(-4,2)', 21, 'f661', 1, 1, 30)
+
+        # normal target
+        observation_simulation(
+            target=target_example,
+            skybg=21,
+            expt=1,
+            nframe=1,
+            band='f661',
+            emgain=30,
+            obsid=51012345678,
+        )
+
+        # bias-gain
+        observation_simulation(
+            target={},
+            skybg=None,
+            expt=1,
+            nframe=1,
+            band='f661',
+            emgain=30,
+            obsid=50112345678,
+        )
+
+
+if __name__ == '__main__':
+    unittest.main()