Coverage for tests/test_matrixMap.py : 24%

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
"""Test MatrixMap constructed with a diagonal vector"""
mm = ast.MatrixMap([-1.0, 2.0]) self.assertEqual(mm.className, "MatrixMap") self.assertEqual(mm.nIn, 2) self.assertEqual(mm.nOut, 2) self.assertTrue(mm.hasForward) self.assertTrue(mm.hasInverse)
self.checkBasicSimplify(mm) self.checkCopy(mm)
indata = np.array([ [1.0, 2.0, 3.0], [0.0, 1.0, 2.0], ], dtype=float) outdata = mm.applyForward(indata) pred_outdata = np.array([ [-1.0, -2.0, -3.0], [0.0, 2.0, 4.0], ], dtype=float) assert_allclose(outdata, pred_outdata)
self.checkRoundTrip(mm, indata) self.checkMappingPersistence(mm, indata)
"""Test MatrixMap constructed with a 2-d matrix
matrix inputs expected outputs 0, 1 (1, 0) (0, 2, -1) 2, 3 (2, 1) (1, 7, -4) -1, -2 (3, 2) (2, 12, -7) """ matrix = np.array([ [0.0, 1.0], [2.0, 3.0], [-1.0, -2.0] ], dtype=float) mm = ast.MatrixMap(matrix) self.assertEqual(mm.nIn, 2) self.assertEqual(mm.nOut, 3) self.assertTrue(mm.hasForward) self.assertFalse(mm.hasInverse)
indata = np.array([ [1.0, 2.0, 3.0], [0.0, 1.0, 2.0], ], dtype=float) outdata = mm.applyForward(indata) pred_outdata = np.array([ [0.0, 1.0, 2.0], [2.0, 7.0, 12.0], [-1.0, -4.0, -7.0], ], dtype=float) assert_allclose(outdata, pred_outdata)
self.checkMappingPersistence(mm, indata)
"""Test that a MatrixMap all coefficients 0 can be simplified
This is ticket DM-10942 """ mm = ast.MatrixMap([0.0, 0.0])
indata = np.array([ [1.0, 2.0, 3.0], [0.0, 1.0, 2.0], ], dtype=float) outdata = mm.applyForward(indata) pred_outdata = np.array([ [0.0, 0.0, 0.0], [0.0, 0.0, 0.0], ], dtype=float) assert_allclose(outdata, pred_outdata)
simplifiedMM = mm.simplified() outdata2 = simplifiedMM.applyForward(indata) assert_allclose(outdata2, pred_outdata)
unittest.main() |