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 FieldsDatabase 

4 

5class TestFieldDatabase(unittest.TestCase): 

6 

7 def setUp(self): 

8 self.fields_db = FieldsDatabase() 

9 self.query = "select * from Field limit 2;" 

10 

11 def test_basic_information_after_creation(self): 

12 self.assertEqual(self.fields_db.db_name, "Fields.db") 

13 self.assertIsNotNone(self.fields_db.connect) 

14 

15 def test_opsim3_userregions(self): 

16 result = self.fields_db.get_opsim3_userregions(self.query) 

17 truth_result = """userRegion = 0.00,-90.00,0.03 

18userRegion = 180.00,-87.57,0.03""" 

19 self.assertEqual(result, truth_result) 

20 

21 def test_get_ra_dec_arrays(self): 

22 ra, dec = self.fields_db.get_ra_dec_arrays(self.query) 

23 self.assertEqual(ra.size, 2) 

24 self.assertEqual(dec.size, 2) 

25 self.assertEqual(ra[1], 180.0) 

26 self.assertAlmostEqual(dec[1], -87.57, delta=1e-2) 

27 

28 def test_get_id_ra_dec_arrays(self): 

29 fieldid, ra, dec = self.fields_db.get_id_ra_dec_arrays(self.query) 

30 self.assertEqual(ra[1], 180.0) 

31 self.assertEqual(fieldid[1], 2) 

32 self.assertNotEqual(fieldid[0], 0) 

33 self.assertAlmostEqual(dec[1], -87.57, delta=1e-2) 

34 

35 def test_get_rows(self): 

36 rows = self.fields_db.get_rows(self.query) 

37 self.assertIsInstance(rows, list) 

38 self.assertEqual(len(rows), 2) 

39 self.assertEqual(len(rows[0]), 8) 

40 

41 def test_get_field_set(self): 

42 field_set = self.fields_db.get_field_set(self.query) 

43 truth_set = set() 

44 truth_set.add((1, 3.5, 0.0, -90.0, -57.068082, -27.128251, 

45 -89.93121, -66.561358)) 

46 truth_set.add((2, 3.5, 180.0, -87.568555, -57.663825, -24.756541, 

47 -96.024547, -66.442665)) 

48 self.assertEqual(len(field_set), 2) 

49 self.assertSetEqual(field_set, truth_set) 

50 

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

52 unittest.main()