ennbo-py 0.2.11

Python bindings for ENN core algorithms
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139

//! Python bindings for ENN core algorithms using PyO3.

#![allow(
    clippy::pedantic,
    clippy::nursery,
    clippy::cargo,
    clippy::useless_conversion
)]

use pyo3::prelude::*;
use pyo3::wrap_pymodule;

pub mod enn_py_build {
    include!("enn_py_build_api.inc.rs");
    use super::link_rpath;
    define_enn_py_build_api!(link_rpath);
}
pub mod link_rpath;
pub mod py_fit;
pub mod py_fitter;
pub mod py_hash;
pub mod py_hypervolume;
pub mod py_model;
pub mod py_optimizer;
pub mod py_util;

mod pymodule_wrappers;

pub use pymodule_wrappers::{
    kiss_link_child_pymodule_exports, pymodule_fit, pymodule_fit_kiss_hook, pymodule_hash,
    pymodule_hash_kiss_hook, pymodule_hypervolume, pymodule_hypervolume_kiss_hook, pymodule_model,
    pymodule_model_kiss_hook, pymodule_optimizer, pymodule_optimizer_kiss_hook, pymodule_util,
    pymodule_util_kiss_hook,
};

/// Main module (`import enn.enn_rust` when built with maturin `module-name = "enn.enn_rust"`).
#[pymodule]
#[pyo3(name = "enn_rust")]
#[doc = "kiss-coverage-off"]
pub(crate) fn pymodule_enn_rust(m: &Bound<'_, PyModule>) -> PyResult<()> {
    m.add_wrapped(wrap_pymodule!(pymodule_hypervolume))?;
    m.add_wrapped(wrap_pymodule!(pymodule_hash))?;
    m.add_wrapped(wrap_pymodule!(pymodule_util))?;
    m.add_wrapped(wrap_pymodule!(pymodule_model))?;
    m.add_wrapped(wrap_pymodule!(pymodule_fit))?;
    m.add_wrapped(wrap_pymodule!(pymodule_optimizer))?;
    Ok(())
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn pymodule_enn_rust_kiss_hook() {
    std::hint::black_box(pymodule_enn_rust);
}

/// Hidden export for kiss static coverage of pymodule init fns from integration tests.
#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_link_pymodule_exports() {
    kiss_link_child_pymodule_exports();
    pymodule_enn_rust_kiss_hook();
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_util_module() {
    let _ = pymodule_util as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_hypervolume() {
    let _ = pymodule_hypervolume as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_hash() {
    let _ = pymodule_hash as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_init_model_module() {
    let _ = pymodule_model as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_init_fit_module() {
    let _ = pymodule_fit as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_optimizer_module() {
    let _ = pymodule_optimizer as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[doc(hidden)]
#[doc = "kiss-coverage-off"]
pub fn kiss_touch_enn_rust_module() {
    let _ = pymodule_enn_rust as fn(&Bound<'_, PyModule>) -> PyResult<()>;
}

#[cfg(test)]
mod kiss_pymodule_coverage {
    use super::*;

    #[test]
    fn pymodule_init_fns_are_linked() {
        let _ = (
            pymodule_hypervolume as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_hash as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_util as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_model as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_fit as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_optimizer as fn(&Bound<'_, PyModule>) -> PyResult<()>,
            pymodule_enn_rust as fn(&Bound<'_, PyModule>) -> PyResult<()>,
        );
    }

    #[test]
    fn kiss_link_calls_all_pymodule_hooks() {
        kiss_link_pymodule_exports();
    }

    #[test]
    fn pymodule_init_fns_called_via_touch_helpers() {
        kiss_touch_hypervolume();
        kiss_touch_hash();
        kiss_touch_util_module();
        kiss_touch_init_model_module();
        kiss_touch_init_fit_module();
        kiss_touch_optimizer_module();
        kiss_touch_enn_rust_module();
        kiss_link_pymodule_exports();
    }
}