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

import unittest 

 

import numpy as np 

from numpy.testing import assert_equal 

 

import astshim as ast 

from astshim.test import ObjectTestCase 

 

 

class TestBase(ObjectTestCase): 

 

def test_arrayFromVector(self): 

nAxes = 3 

nValues = 5 

np.random.seed(1) 

dataVec = np.random.rand(nAxes * nValues) 

desiredDataArr = dataVec.copy() 

desiredDataArr.shape = (nAxes, nValues) 

dataArr = ast.arrayFromVector(vec=dataVec, nAxes=nAxes) 

assert_equal(dataArr, desiredDataArr) 

 

dataArr2 = ast.arrayFromVector(vec=list(dataVec), nAxes=nAxes) 

assert_equal(dataArr2, desiredDataArr) 

 

# make sure dataArr is a deep copy; changing dataVec should 

# not change dataArr 

dataVec[0] += 10 

assert_equal(dataArr, desiredDataArr) 

 

for delta in (-1, 1): 

badDataVec = np.random.rand(nAxes * nValues + delta) 

with self.assertRaises(RuntimeError): 

ast.arrayFromVector(vec=badDataVec, nAxes=nAxes) 

 

 

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

unittest.main()