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;
 | 
						|
    });
 | 
						|
}
 |