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

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

import unittest 

 

import numpy as np 

from numpy.testing import assert_allclose 

 

import astshim as ast 

from astshim.test import MappingTestCase 

 

 

class TestRateMap(MappingTestCase): 

 

def test_RateMapBasics(self): 

zoomfac = 0.523 

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

self.assertIsInstance(ratemap, ast.RateMap) 

self.assertIsInstance(ratemap, ast.Mapping) 

self.assertEqual(ratemap.nOut, 1) 

 

self.checkBasicSimplify(ratemap) 

self.checkCopy(ratemap) 

 

indata = np.array([ 

[1.1, -43.5, -5.54], 

[2.2, 1309.31, 35.2], 

]) 

outdata = ratemap.applyForward(indata) 

assert_allclose(outdata, zoomfac) 

 

self.checkMappingPersistence(ratemap, indata) 

 

def test_RateMap2(self): 

zoomfac = 23.323 

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

 

indata = np.array([ 

[1.1, -43.5, -5.54], 

[2.2, 1309.31, 35.2], 

]) 

outdata = ratemap.applyForward(indata) 

assert_allclose(outdata, 0) 

 

self.checkMappingPersistence(ratemap, indata) 

 

 

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

unittest.main()