Merge pull request #48 from bjodah/atol-length-check

Check atol vector length
bjodah · Jan 31, 2018 · 8f1eab7 · 8f1eab7
2 parents 75b5492 + c503879
commit 8f1eab7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pycvodes/include/cvodes_anyode.hpp b/pycvodes/include/cvodes_anyode.hpp
@@ -195,7 +195,9 @@ namespace cvodes_anyode {
             integr.root_init(nroots, roots_cb<OdeSys>);
         if (atol.size() == 1){
             integr.set_tol(rtol, atol[0]);
-        }else{
+        } else if (atol.size() != (size_t)ny) {
+            throw std::runtime_error("atol of incorrect length");
+        } else {
             integr.set_tol(rtol, atol);
         }
         integr.set_init_step(dx0);