Coverage for tests/test_base.py: 31%
Shortcuts 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
Shortcuts 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 unittest
3import numpy as np
4from numpy.testing import assert_equal
6import astshim as ast
7from astshim.test import ObjectTestCase
10class TestBase(ObjectTestCase):
12 def test_arrayFromVector(self):
13 nAxes = 3
14 nValues = 5
15 np.random.seed(1)
16 dataVec = np.random.rand(nAxes * nValues)
17 desiredDataArr = dataVec.copy()
18 desiredDataArr.shape = (nAxes, nValues)
19 dataArr = ast.arrayFromVector(vec=dataVec, nAxes=nAxes)
20 assert_equal(dataArr, desiredDataArr)
22 dataArr2 = ast.arrayFromVector(vec=list(dataVec), nAxes=nAxes)
23 assert_equal(dataArr2, desiredDataArr)
25 # make sure dataArr is a deep copy; changing dataVec should
26 # not change dataArr
27 dataVec[0] += 10
28 assert_equal(dataArr, desiredDataArr)
30 for delta in (-1, 1):
31 badDataVec = np.random.rand(nAxes * nValues + delta)
32 with self.assertRaises(RuntimeError):
33 ast.arrayFromVector(vec=badDataVec, nAxes=nAxes)
35 def testVersion(self):
36 version = ast.astVersion()
37 self.assertGreaterEqual(version, 9001000)
40if __name__ == "__main__": 40 ↛ 41line 40 didn't jump to line 41, because the condition on line 40 was never true
41 unittest.main()