Coverage for tests/test_rateMap.py: 34%

27 statements  

« prev     ^ index     » next       coverage.py v7.2.3, created at 2023-04-19 10:51 +0000

1import unittest 

2 

3import numpy as np 

4from numpy.testing import assert_allclose 

5 

6import astshim as ast 

7from astshim.test import MappingTestCase 

8 

9 

10class TestRateMap(MappingTestCase): 

11 

12 def test_RateMapBasics(self): 

13 zoomfac = 0.523 

14 ratemap = ast.RateMap(ast.ZoomMap(2, zoomfac), 2, 2) 

15 self.assertIsInstance(ratemap, ast.RateMap) 

16 self.assertIsInstance(ratemap, ast.Mapping) 

17 self.assertEqual(ratemap.nOut, 1) 

18 

19 self.checkBasicSimplify(ratemap) 

20 self.checkCopy(ratemap) 

21 

22 indata = np.array([ 

23 [1.1, -43.5, -5.54], 

24 [2.2, 1309.31, 35.2], 

25 ]) 

26 outdata = ratemap.applyForward(indata) 

27 assert_allclose(outdata, zoomfac) 

28 

29 self.checkMappingPersistence(ratemap, indata) 

30 

31 def test_RateMap2(self): 

32 zoomfac = 23.323 

33 ratemap = ast.RateMap(ast.ZoomMap(2, zoomfac), 2, 1) 

34 

35 indata = np.array([ 

36 [1.1, -43.5, -5.54], 

37 [2.2, 1309.31, 35.2], 

38 ]) 

39 outdata = ratemap.applyForward(indata) 

40 assert_allclose(outdata, 0) 

41 

42 self.checkMappingPersistence(ratemap, indata) 

43 

44 

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

46 unittest.main()