Python selective_func Beispiele

Beispiel #1

Datei: test_numpy_vectorize.py Projekt: jcelerier/pybind11

def test_type_selection():
    from pybind11_tests import selective_func

    assert selective_func(np.array([1], dtype=np.int32)) == "Int branch taken."
    assert selective_func(np.array([1.0],
                                   dtype=np.float32)) == "Float branch taken."
    assert selective_func(np.array(
        [1.0j], dtype=np.complex64)) == "Complex float branch taken."

Beispiel #2

Datei: test_numpy_vectorize.py Projekt: MortezaRamezani/gem5

def test_type_selection():
    from pybind11_tests import selective_func

    assert selective_func(np.array([1], dtype=np.int32)) == "Int branch taken."
    assert selective_func(np.array([1.0], dtype=np.float32)) == "Float branch taken."
    assert selective_func(np.array([1.0j], dtype=np.complex64)) == "Complex float branch taken."