def testNone(self): array = numpy.zeros(10, dtype=float) self.assertEqual(pybind11_test_mod.acceptNoneArray(array), 0) self.assertEqual(pybind11_test_mod.acceptNoneArray(None), 1) self.assertEqual(pybind11_test_mod.acceptNoneArray(), 1) m1 = pybind11_test_mod.returnMatrixXd() self.assertEqual(pybind11_test_mod.acceptNoneMatrixXd(m1), 2) self.assertEqual(pybind11_test_mod.acceptNoneMatrixXd(None), 3) self.assertEqual(pybind11_test_mod.acceptNoneMatrixXd(), 3) m2 = pybind11_test_mod.returnMatrix2d() self.assertEqual(pybind11_test_mod.acceptNoneMatrix2d(m2), 4) self.assertEqual(pybind11_test_mod.acceptNoneMatrix2d(None), 5) self.assertEqual(pybind11_test_mod.acceptNoneMatrix2d(), 5)
def testMatrixXd(self): m1 = pybind11_test_mod.returnMatrixXd() m2 = numpy.matrix( numpy.arange(15, dtype=float).reshape(3, 5).transpose()) self.assert_((m1 == m2).all()) self.assert_(pybind11_test_mod.acceptMatrixXd(m2))
def testMatrixXd(self): m1 = pybind11_test_mod.returnMatrixXd() m2 = numpy.matrix(numpy.arange(15, dtype=float).reshape(3,5).transpose()) self.assert_((m1 == m2).all()) self.assert_(pybind11_test_mod.acceptMatrixXd(m2))