First commit of Polymath

tests/tests.cpp

@@ -0,0 +1,50 @@
+#include <string>
+#include <locale>
+#include "tests.h"
+
+
+int RunTests()
+{
+//    Py_SetProgramName("MatricesTest");  /* optional but recommended */
+    Py_Initialize();
+    PyRun_SimpleString("import numpy as np");
+    PyRun_SimpleString("def compare_floats(f1,f2,tol):\n    return np.isclose(f1,f2,rtol=tol)");
+    bool print = 0;
+    for(int i = 0; i < 10; i++)
+    {
+//        std::cout<<TestInverse<float>(2,print)<<std::endl;
+
+        for(long j = 2; j < 20; j++)
+        {
+            if(!TestInverse<float>(j,print))
+            {
+                std::cerr<<"Error testing float inverse. Try again to verify that this is not a statistical error."<<std::endl;
+                std::exit(1);
+            }
+            if(!TestInverse<double>(j,print))
+            {
+                std::cerr<<"Error testing double inverse. Try again to verify that this is not a statistical error."<<std::endl;
+                std::exit(1);
+            }
+            if(!TestInverse<std::complex<float>>(j,print))
+            {
+                std::cerr<<"Error testing complex float inverse. Try again to verify that this is not a statistical error."<<std::endl;
+                std::exit(1);
+            }
+            if(!TestInverse<std::complex<double>>(j,print))
+            {
+                std::cerr<<"Error testing complex double inverse. Try again to verify that this is not a statistical error."<<std::endl;
+                std::exit(1);
+            }
+        }
+    }
+    Py_Finalize();
+    return 0;
+}
+
+int main()
+{
+    RunTests();
+    std::cout<<"Tests program exited with no errors."<<std::endl;
+    return 0;
+}

tests/tests.h

@@ -0,0 +1,84 @@
+#ifndef TESTS_H
+#define TESTS_H
+
+#include "Polymath.h"
+#include <Python.h>
+
+template <typename T>
+int precision_per_type()
+{
+    if(std::is_same<T,double>::value || std::is_same<T,std::complex<double>>::value)
+        return 10;
+    else if(std::is_same<T,float>::value || std::is_same<T,std::complex<float>>::value)
+        return 2;
+    else
+        return 2;
+}
+
+template <typename T>
+std::string python_type_per_type()
+{
+    return std::string("np.complex128");
+    if(std::is_same<T,double>::value)
+    {
+        return std::string("np.double");
+    }
+    else if(std::is_same<T,std::complex<double>>::value)
+    {
+        return std::string("np.complex128");
+    }
+    else if(std::is_same<T,float>::value)
+    {
+        return std::string("np.float32");
+    }
+    else if(std::is_same<T,std::complex<float>>::value)
+    {
+        return std::string("np.complex64");
+    }
+    else
+    {
+        return std::string("np.complex128");
+    }
+}
+
+template <typename T>
+bool TestInverse(int len, bool print)
+{
+    Poly::Matrix<T> mat_d = Poly::RandomMatrix<T>(len,len,0,10,std::random_device{}());
+    if(print) std::cout<<mat_d<<std::endl;
+    if(print) std::cout<<mat_d.asString(32,'[',']',',')<<std::endl;
+    int prec = precision_per_type<T>();
+    auto mat_d_i = mat_d.getInverse();
+    PyObject *main = PyImport_AddModule("__main__");
+    PyRun_SimpleString(std::string("data={}").c_str());
+    PyRun_SimpleString(std::string("data['a']=np.matrix(" + mat_d.asString(32,'[',']',',') + ",dtype="+python_type_per_type<T>()+")").c_str());
+    PyRun_SimpleString(std::string("data['b_c']=np.matrix(" + mat_d_i.asString(32,'[',']',',') + ",dtype="+std::string("np.complex128")+")").c_str());
+    PyRun_SimpleString(std::string("data['b_p']=np.linalg.inv(data['a'])").c_str());
+    if(print) std::cout << mat_d.columns() << std::endl;
+    if(print) PyRun_SimpleString(std::string("print('Cond num: ', np.linalg.cond(data['a']))").c_str());
+    if(print) PyRun_SimpleString(std::string("print('Cond num: ', np.linalg.cond(data['b_c']))").c_str());
+    if(print) PyRun_SimpleString(std::string("print('Cond num: ', np.linalg.cond(data['b_c']))").c_str());
+    if(print) PyRun_SimpleString("print(data['a'])");
+    if(print) PyRun_SimpleString("print(data['b_c'])");
+    if(print) PyRun_SimpleString("print(data['b_p'])");
+    if(print) PyRun_SimpleString("print(abs(data['b_c']-data['b_p']))");
+    PyRun_SimpleString("data['fb_p']=((data['b_p']).flatten().tolist())[0]");
+    PyRun_SimpleString("data['fb_c']=((data['b_c']).flatten().tolist())[0]");
+
+    PyRun_SimpleString(std::string("res=list(set([compare_floats(data['fb_p'][i],data['fb_c'][i],1e-"+ std::to_string(prec) +") for i in range(len(data['fb_p']))]))").c_str());
+    PyRun_SimpleString(std::string("res = ((len(res) == 1) and res[0])").c_str());
+
+    if(print) PyRun_SimpleString("print('Is close: ',res)");
+//    getchar();
+    PyObject *globals = PyModule_GetDict(main);
+    PyObject *a = PyDict_GetItemString(globals, "res");
+    bool ret = PyObject_IsTrue(a);
+    PyRun_SimpleString("del data");
+    PyRun_SimpleString("del res");
+    return ret;
+}
+
+
+int RunTests();
+
+#endif // TESTS_H