test_eval.cpp 2.32 KB
Newer Older
Klemens Morgenstern's avatar
Klemens Morgenstern committed
1
/*
Dean Moldovan's avatar
Dean Moldovan committed
2
    tests/test_eval.cpp -- Usage of eval() and eval_file()
Klemens Morgenstern's avatar
Klemens Morgenstern committed
3
4
5
6
7
8
9
10

    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.
*/


11
#include <pybind11/eval.h>
Dean Moldovan's avatar
Dean Moldovan committed
12
#include "pybind11_tests.h"
Klemens Morgenstern's avatar
Klemens Morgenstern committed
13

14
test_initializer eval([](py::module &m) {
15
16
17
18
19
20
21
    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;
        });
Klemens Morgenstern's avatar
Klemens Morgenstern committed
22

23
        // Regular string literal
24
        py::exec(
25
26
            "message = 'Hello World!'\n"
            "x = call_test()",
27
28
            global, local
        );
29
30

        // Multi-line raw string literal
31
        py::exec(R"(
32
33
34
35
36
37
            if x == 42:
                print(message)
            else:
                raise RuntimeError
            )", global, local
        );
38
        auto x = local["x"].cast<int>();
Klemens Morgenstern's avatar
Klemens Morgenstern committed
39

40
        return x == 42;
41
    });
Klemens Morgenstern's avatar
Klemens Morgenstern committed
42

43
44
45
46
47
48
    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;
    });
Klemens Morgenstern's avatar
Klemens Morgenstern committed
49

50
51
52
53
54
    m.def("test_eval_single_statement", []() {
        auto local = py::dict();
        local["call_test"] = py::cpp_function([&]() -> int {
            return 42;
        });
Klemens Morgenstern's avatar
Klemens Morgenstern committed
55

56
57
        auto result = py::eval<py::eval_single_statement>("x = call_test()", py::dict(), local);
        auto x = local["x"].cast<int>();
58
        return result.is_none() && x == 42;
59
60
61
62
63
64
65
66
67
68
    });

    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);
69
        return val_out == 43 && result.is_none();
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
    });

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