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

1import unittest 

2 

3from lsst.sims.survey.fields import FieldSelection 

4 

5class TestFieldSelection(unittest.TestCase): 

6 

7 def setUp(self): 

8 self.fs = FieldSelection() 

9 

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' 

20 

21 def test_base_select(self): 

22 self.assertEqual(self.fs.base_select(), self.truth_base_query) 

23 

24 def test_finish_query(self): 

25 query = "silly query" 

26 self.assertEqual(self.fs.finish_query(query), query + ";") 

27 

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) 

34 

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) 

40 

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",) 

45 

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") 

52 

53 truth_query = " ".join(truth_query_parts) + ";" 

54 self.assertEqual(self.fs.combine_queries(query1, query2, 

55 combiners=combiners), 

56 truth_query) 

57 

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) 

63 

64 combiners = ("and",) 

65 with self.assertRaises(RuntimeError): 

66 self.fs.combine_queries(query1, combiners=combiners) 

67 

68 def test_get_all_fields(self): 

69 self.assertEqual(self.fs.get_all_fields(), 

70 self.truth_base_query + ";") 

71 

72 def test_get_user_regions(self): 

73 user_regions = (2, 256, 2935) 

74 query = self.fs.select_user_regions(user_regions) 

75 combiners = () 

76 

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") 

81 

82 truth_query = " ".join(truth_query_parts) + ";" 

83 self.assertEqual(self.fs.combine_queries(query, combiners=combiners), 

84 truth_query) 

85 

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

87 unittest.main()