Coverage for tests/test_ingestion.py: 58%

49 statements  

« prev     ^ index     » next       coverage.py v7.5.0, created at 2024-05-02 05:06 -0700

1"""Unit tests for Gen3 RubinGC raw data ingest. 

2""" 

3 

4import unittest 

5import os 

6import lsst.utils.tests 

7 

8from lsst.obs.base.ingest_tests import IngestTestBase 

9from lsst.obs.rubinGenericCamera import StarTrackerWide, StarTrackerNarrow, StarTrackerFast 

10from lsst.obs.rubinGenericCamera.filters import RUBIN_GENERIC_CAMERA_FILTER_DEFINITIONS 

11 

12testDataPackage = "obs_rubinGenericCamera" 

13try: 

14 testDataDirectory = lsst.utils.getPackageDir(testDataPackage) 

15except (LookupError, lsst.pex.exceptions.NotFoundError): 

16 testDataDirectory = None 

17 

18 

19@unittest.skipIf(testDataDirectory is None, "obs_rubinGenericCamera must be set up") 

20class StarTrackerNarrowIngestTestCase(IngestTestBase, lsst.utils.tests.TestCase): 

21 instrumentClassName = "lsst.obs.rubinGenericCamera.StarTrackerNarrow" 

22 

23 visits = None # we don't have a definition of visits 

24 

25 def setUp(self): 

26 self.ingestdir = os.path.dirname(__file__) 

27 self.instrument = StarTrackerNarrow() 

28 self.file = os.path.join(testDataDirectory, "data", "input", "raw", "GC102_O_20221208_000211.fits.gz") 

29 self.dataIds = [dict(instrument="StarTrackerNarrow", exposure=2022120800211, detector=0)] 

30 self.filterLabel = RUBIN_GENERIC_CAMERA_FILTER_DEFINITIONS[0].makeFilterLabel() 

31 

32 super().setUp() 

33 

34 

35@unittest.skipIf(testDataDirectory is None, "obs_rubinGenericCamera must be set up") 

36class StarTrackerWideIngestTestCase(IngestTestBase, lsst.utils.tests.TestCase): 

37 instrumentClassName = "lsst.obs.rubinGenericCamera.StarTrackerWide" 

38 

39 visits = None # we don't have a definition of visits 

40 

41 def setUp(self): 

42 self.ingestdir = os.path.dirname(__file__) 

43 self.instrument = StarTrackerWide() 

44 self.file = os.path.join(testDataDirectory, "data", "input", "raw", "GC101_O_20221208_000211.fits.gz") 

45 self.dataIds = [dict(instrument="StarTrackerWide", exposure=2022120800211, detector=0)] 

46 self.filterLabel = RUBIN_GENERIC_CAMERA_FILTER_DEFINITIONS[0].makeFilterLabel() 

47 

48 super().setUp() 

49 

50 

51@unittest.skipIf(testDataDirectory is None, "obs_rubinGenericCamera must be set up") 

52class StarTrackerFastIngestTestCase(IngestTestBase, lsst.utils.tests.TestCase): 

53 instrumentClassName = "lsst.obs.rubinGenericCamera.StarTrackerFast" 

54 

55 visits = None # we don't have a definition of visits 

56 

57 def setUp(self): 

58 self.ingestdir = os.path.dirname(__file__) 

59 self.instrument = StarTrackerFast() 

60 self.file = os.path.join(testDataDirectory, "data", "input", "raw", "GC103_O_20221208_000211.fits.gz") 

61 self.dataIds = [dict(instrument="StarTrackerFast", exposure=2022120800211, detector=0)] 

62 self.filterLabel = RUBIN_GENERIC_CAMERA_FILTER_DEFINITIONS[0].makeFilterLabel() 

63 

64 super().setUp() 

65 

66 

67def setup_module(module): 

68 lsst.utils.tests.init() 

69 

70 

71if __name__ == "__main__": 71 ↛ 72line 71 didn't jump to line 72, because the condition on line 71 was never true

72 lsst.utils.tests.init() 

73 unittest.main()