Hide keyboard shortcuts

Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

# This file is part of dax_ppdb. 

# 

# Developed for the LSST Data Management System. 

# This product includes software developed by the LSST Project 

# (https://www.lsst.org). 

# See the COPYRIGHT file at the top-level directory of this distribution 

# for details of code ownership. 

# 

# This program is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License as published by 

# the Free Software Foundation, either version 3 of the License, or 

# (at your option) any later version. 

# 

# This program is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the GNU General Public License 

# along with this program. If not, see <https://www.gnu.org/licenses/>. 

 

import unittest.mock 

import lsst.utils.tests 

 

import lsst.afw.table as afwTable 

import lsst.afw.image as afwImage 

import lsst.geom as geom 

import lsst.daf.base as dafBase 

from lsst.dax.ppdb import Ppdb, PpdbConfig 

 

 

def createTestObjects(n_objects, extra_fields): 

"""Create test objects to store in the Ppdb. 

 

Parameters 

---------- 

n_objects : `int` 

Number of objects to create. 

extra_fields : `dict` 

A `dict` whose keys are field names and whose values are their types. 

 

Returns 

------- 

sources : `lsst.afw.table.SourceCatalog` 

Tests sources with filled values. 

""" 

schema = afwTable.SourceTable.makeMinimalSchema() 

for field, type in extra_fields.items(): 

schema.addField(field, type=type) 

sources = afwTable.SourceCatalog(schema) 

 

for src_idx in range(n_objects): 

src = sources.addNew() 

for subSchema in schema: 

if subSchema.getField().getTypeString() == "Angle": 

src[subSchema.getField().getName()] = 1 * geom.degrees 

else: 

src[subSchema.getField().getName()] = 1 

src['id'] = src_idx 

 

return sources 

 

 

class TestApVerifyQueries(unittest.TestCase): 

 

def setUp(self): 

self.ppdbCfg = PpdbConfig() 

# Create DB in memory. 

self.ppdbCfg.db_url = 'sqlite://' 

self.ppdbCfg.isolation_level = "READ_UNCOMMITTED" 

self.ppdbCfg.dia_object_index = "baseline" 

self.ppdbCfg.dia_object_columns = [] 

self.ppdb = Ppdb( 

config=self.ppdbCfg, 

afw_schemas=dict(DiaObject=afwTable.SourceTable.makeMinimalSchema(), 

DiaSource=afwTable.SourceTable.makeMinimalSchema())) 

self.ppdb._schema.makeSchema() 

 

def tearDown(self): 

del self.ppdb 

 

def test_count_zero_objects(self): 

value = self.ppdb.countUnassociatedObjects() 

self.assertEqual(value, 0) 

 

def test_count_objects(self): 

n_created = 5 

sources = createTestObjects(n_created, {'nDiaSources': 'I'}) 

sources[-1]['nDiaSources'] = 2 

 

# nsecs must be an integer, not 1.4e18 

dateTime = dafBase.DateTime(nsecs=1400000000 * 10**9) 

self.ppdb.storeDiaObjects(sources, dateTime.toPython()) 

 

value = self.ppdb.countUnassociatedObjects() 

self.assertEqual(n_created - 1, value) 

 

@staticmethod 

def _makeVisitInfo(exposureId): 

# Real VisitInfo hard to create 

visitInfo = unittest.mock.NonCallableMock( 

afwImage.VisitInfo, 

**{"getExposureId.return_value": exposureId} 

) 

return visitInfo 

 

def test_isExposureProcessed(self): 

n_created = 5 

sources = createTestObjects(n_created, {'ccdVisitId': 'I'}) 

for source in sources: 

source['ccdVisitId'] = 2381 

 

self.ppdb.storeDiaSources(sources) 

 

self.assertTrue(self.ppdb.isVisitProcessed(TestApVerifyQueries._makeVisitInfo(2381))) 

self.assertFalse(self.ppdb.isVisitProcessed(TestApVerifyQueries._makeVisitInfo(42))) 

 

 

class MemoryTester(lsst.utils.tests.MemoryTestCase): 

pass 

 

 

def setup_module(module): 

lsst.utils.tests.init() 

 

 

127 ↛ 128line 127 didn't jump to line 128, because the condition on line 127 was never trueif __name__ == "__main__": 

lsst.utils.tests.init() 

unittest.main()