Coverage for tests/test_BasisFuncs.py : 41%

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
9class TestBasis(unittest.TestCase):
11 def testVisit_repeat_basis_function(self):
12 bf = basis_functions.Visit_repeat_basis_function()
14 indx = np.array([1000])
16 # 30 minute step
17 delta = 30./60./24.
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.)
28 # Advance time so now we want a pair
29 conditions.mjd += delta
30 self.assertEqual(np.max(bf(conditions)), 1.)
32 # Now complete the pair and it should go back to zero
33 bf.add_observation(obs, indx=indx)
35 conditions.mjd += delta
36 self.assertEqual(np.max(bf(conditions)), 0.)
39class TestMemory(lsst.utils.tests.MemoryTestCase):
40 pass
43def setup_module(module):
44 lsst.utils.tests.init()
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()