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 numpy as np 

2import unittest 

3import lsst.utils.tests 

4import lsst.sims.featureScheduler.basis_functions as basis_functions 

5from lsst.sims.featureScheduler.utils import empty_observation 

6from lsst.sims.featureScheduler.features import Conditions 

7 

8 

9class TestBasis(unittest.TestCase): 

10 

11 def testVisit_repeat_basis_function(self): 

12 bf = basis_functions.Visit_repeat_basis_function() 

13 

14 indx = np.array([1000]) 

15 

16 # 30 minute step 

17 delta = 30./60./24. 

18 

19 # Add 1st observation, should still be zero 

20 obs = empty_observation() 

21 obs['filter'] = 'r' 

22 obs['mjd'] = 59000. 

23 conditions = Conditions() 

24 conditions.mjd = np.max(obs['mjd']) 

25 bf.add_observation(obs, indx=indx) 

26 self.assertEqual(np.max(bf(conditions)), 0.) 

27 

28 # Advance time so now we want a pair 

29 conditions.mjd += delta 

30 self.assertEqual(np.max(bf(conditions)), 1.) 

31 

32 # Now complete the pair and it should go back to zero 

33 bf.add_observation(obs, indx=indx) 

34 

35 conditions.mjd += delta 

36 self.assertEqual(np.max(bf(conditions)), 0.) 

37 

38 

39class TestMemory(lsst.utils.tests.MemoryTestCase): 

40 pass 

41 

42 

43def setup_module(module): 

44 lsst.utils.tests.init() 

45 

46 

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

48 lsst.utils.tests.init() 

49 unittest.main()