92 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			C++
		
	
	
| /*
 | |
|     tests/test_eval.cpp -- Usage of eval() and eval_file()
 | |
| 
 | |
|     Copyright (c) 2016 Klemens D. Morgenstern
 | |
| 
 | |
|     All rights reserved. Use of this source code is governed by a
 | |
|     BSD-style license that can be found in the LICENSE file.
 | |
| */
 | |
| 
 | |
| 
 | |
| #include <pybind11/eval.h>
 | |
| #include "pybind11_tests.h"
 | |
| 
 | |
| TEST_SUBMODULE(eval_, m) {
 | |
|     // test_evals
 | |
| 
 | |
|     auto global = py::dict(py::module::import("__main__").attr("__dict__"));
 | |
| 
 | |
|     m.def("test_eval_statements", [global]() {
 | |
|         auto local = py::dict();
 | |
|         local["call_test"] = py::cpp_function([&]() -> int {
 | |
|             return 42;
 | |
|         });
 | |
| 
 | |
|         // Regular string literal
 | |
|         py::exec(
 | |
|             "message = 'Hello World!'\n"
 | |
|             "x = call_test()",
 | |
|             global, local
 | |
|         );
 | |
| 
 | |
|         // Multi-line raw string literal
 | |
|         py::exec(R"(
 | |
|             if x == 42:
 | |
|                 print(message)
 | |
|             else:
 | |
|                 raise RuntimeError
 | |
|             )", global, local
 | |
|         );
 | |
|         auto x = local["x"].cast<int>();
 | |
| 
 | |
|         return x == 42;
 | |
|     });
 | |
| 
 | |
|     m.def("test_eval", [global]() {
 | |
|         auto local = py::dict();
 | |
|         local["x"] = py::int_(42);
 | |
|         auto x = py::eval("x", global, local);
 | |
|         return x.cast<int>() == 42;
 | |
|     });
 | |
| 
 | |
|     m.def("test_eval_single_statement", []() {
 | |
|         auto local = py::dict();
 | |
|         local["call_test"] = py::cpp_function([&]() -> int {
 | |
|             return 42;
 | |
|         });
 | |
| 
 | |
|         auto result = py::eval<py::eval_single_statement>("x = call_test()", py::dict(), local);
 | |
|         auto x = local["x"].cast<int>();
 | |
|         return result.is_none() && x == 42;
 | |
|     });
 | |
| 
 | |
|     m.def("test_eval_file", [global](py::str filename) {
 | |
|         auto local = py::dict();
 | |
|         local["y"] = py::int_(43);
 | |
| 
 | |
|         int val_out;
 | |
|         local["call_test2"] = py::cpp_function([&](int value) { val_out = value; });
 | |
| 
 | |
|         auto result = py::eval_file(filename, global, local);
 | |
|         return val_out == 43 && result.is_none();
 | |
|     });
 | |
| 
 | |
|     m.def("test_eval_failure", []() {
 | |
|         try {
 | |
|             py::eval("nonsense code ...");
 | |
|         } catch (py::error_already_set &) {
 | |
|             return true;
 | |
|         }
 | |
|         return false;
 | |
|     });
 | |
| 
 | |
|     m.def("test_eval_file_failure", []() {
 | |
|         try {
 | |
|             py::eval_file("non-existing file");
 | |
|         } catch (std::exception &) {
 | |
|             return true;
 | |
|         }
 | |
|         return false;
 | |
|     });
 | |
| }
 |