Coverage for tests/test_field_selection.py : 24%

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
1import unittest
3from lsst.sims.survey.fields import FieldSelection
5class TestFieldSelection(unittest.TestCase):
7 def setUp(self):
8 self.fs = FieldSelection()
10 self.truth_base_query = "select * from Field"
11 self.truth_galactic_exclusion = '(abs(fieldGB) > (10.0 - '\
12 '(9.9 * abs(fieldGL)) / 90.0))'
13 self.truth_galactic_region = '(abs(fieldGB) <= (10.0 - '\
14 '(9.9 * abs(fieldGL)) / 90.0))'
15 self.truth_normal_ra_region = 'fieldRA between 90.0 and 270.0'
16 self.truth_cross_region = '(fieldRA between 270.0 and 360 or '\
17 'fieldRA between 0 and 90.0)'
18 self.truth_normal_dec_region = 'fieldDec between -90.0 and -61.0'
19 self.truth_user_regions = 'fieldId=2 or fieldId=256 or fieldId=2935'
21 def test_base_select(self):
22 self.assertEqual(self.fs.base_select(), self.truth_base_query)
24 def test_finish_query(self):
25 query = "silly query"
26 self.assertEqual(self.fs.finish_query(query), query + ";")
28 def test_galactic_region(self):
29 self.assertEqual(self.fs.galactic_region(10.0, 0.1, 90.0,
30 exclusion=True),
31 self.truth_galactic_exclusion)
32 self.assertEqual(self.fs.galactic_region(10.0, 0.1, 90.0),
33 self.truth_galactic_region)
35 def test_select_region(self):
36 self.assertEqual(self.fs.select_region("RA", 90.0, 270.0),
37 self.truth_normal_ra_region)
38 self.assertEqual(self.fs.select_region("RA", 270.0, 90.0),
39 self.truth_cross_region)
41 def test_combine_queries(self):
42 query1 = self.fs.select_region("RA", 90.0, 270.0)
43 query2 = self.fs.select_region("Dec", -90.0, -61.0)
44 combiners = ("and",)
46 truth_query_parts = [self.truth_base_query]
47 truth_query_parts.append("where")
48 truth_query_parts.append(self.truth_normal_ra_region)
49 truth_query_parts.append("and")
50 truth_query_parts.append(self.truth_normal_dec_region)
51 truth_query_parts.append("order by fieldId")
53 truth_query = " ".join(truth_query_parts) + ";"
54 self.assertEqual(self.fs.combine_queries(query1, query2,
55 combiners=combiners),
56 truth_query)
58 def test_bad_combine_queries(self):
59 query1 = self.fs.select_region("RA", 90.0, 270.0)
60 query2 = self.fs.select_region("Dec", -90.0, -61.0)
61 with self.assertRaises(RuntimeError):
62 self.fs.combine_queries(query1, query2)
64 combiners = ("and",)
65 with self.assertRaises(RuntimeError):
66 self.fs.combine_queries(query1, combiners=combiners)
68 def test_get_all_fields(self):
69 self.assertEqual(self.fs.get_all_fields(),
70 self.truth_base_query + ";")
72 def test_get_user_regions(self):
73 user_regions = (2, 256, 2935)
74 query = self.fs.select_user_regions(user_regions)
75 combiners = ()
77 truth_query_parts = [self.truth_base_query]
78 truth_query_parts.append("where")
79 truth_query_parts.append(self.truth_user_regions)
80 truth_query_parts.append("order by fieldId")
82 truth_query = " ".join(truth_query_parts) + ";"
83 self.assertEqual(self.fs.combine_queries(query, combiners=combiners),
84 truth_query)
86if __name__ == '__main__': 86 ↛ 87line 86 didn't jump to line 87, because the condition on line 86 was never true
87 unittest.main()