xcoll 0.5.12__py3-none-any.whl → 0.6.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of xcoll might be problematic. Click here for more details.

Files changed (333) hide show
  1. xcoll/__init__.py +5 -18
  2. xcoll/beam_elements/__init__.py +1 -0
  3. xcoll/beam_elements/absorber.py +12 -2
  4. xcoll/beam_elements/base.py +162 -62
  5. xcoll/beam_elements/blowup.py +1 -0
  6. xcoll/beam_elements/elements_src/black_absorber.h +57 -58
  7. xcoll/beam_elements/elements_src/black_crystal.h +49 -50
  8. xcoll/beam_elements/elements_src/everest_block.h +21 -11
  9. xcoll/beam_elements/elements_src/everest_collimator.h +100 -91
  10. xcoll/beam_elements/elements_src/everest_crystal.h +145 -140
  11. xcoll/beam_elements/elements_src/transparent_collimator.h +126 -0
  12. xcoll/beam_elements/elements_src/transparent_crystal.h +118 -0
  13. xcoll/beam_elements/everest.py +16 -5
  14. xcoll/beam_elements/monitor.py +1 -0
  15. xcoll/beam_elements/transparent.py +83 -0
  16. xcoll/colldb.py +15 -6
  17. xcoll/general.py +1 -1
  18. xcoll/headers/particle_states.py +51 -0
  19. xcoll/initial_distribution.py +121 -91
  20. xcoll/interaction_record/interaction_record.py +2 -1
  21. xcoll/interaction_record/interaction_types.py +2 -2
  22. xcoll/line_tools.py +163 -73
  23. xcoll/lossmap.py +519 -127
  24. xcoll/plot.py +109 -0
  25. xcoll/rf_sweep.py +6 -0
  26. xcoll/scattering_routines/engine.py +414 -217
  27. xcoll/scattering_routines/environment.py +297 -0
  28. xcoll/scattering_routines/everest/amorphous.h +95 -71
  29. xcoll/scattering_routines/everest/{channeling.h → channelling.h} +121 -112
  30. xcoll/scattering_routines/everest/constants.h +1 -1
  31. xcoll/scattering_routines/everest/crystal_parameters.h +9 -9
  32. xcoll/scattering_routines/everest/everest.h +8 -3
  33. xcoll/scattering_routines/everest/everest.py +2 -1
  34. xcoll/scattering_routines/everest/ionisation_loss.h +141 -0
  35. xcoll/scattering_routines/everest/jaw.h +19 -24
  36. xcoll/scattering_routines/everest/materials.py +2 -0
  37. xcoll/scattering_routines/everest/multiple_coulomb_scattering.h +2 -2
  38. xcoll/scattering_routines/everest/nuclear_interaction.h +35 -19
  39. xcoll/scattering_routines/everest/properties.h +3 -72
  40. xcoll/xaux.py +65 -109
  41. {xcoll-0.5.12.dist-info → xcoll-0.6.0.dist-info}/METADATA +5 -5
  42. xcoll-0.6.0.dist-info/RECORD +135 -0
  43. xcoll/_manager.py +0 -22
  44. xcoll/headers/particle_states.h +0 -25
  45. xcoll/install.py +0 -35
  46. xcoll/scattering_routines/geant4/collimasim/.git +0 -1
  47. xcoll/scattering_routines/geant4/collimasim/.gitignore +0 -12
  48. xcoll/scattering_routines/geant4/collimasim/.gitmodules +0 -3
  49. xcoll/scattering_routines/geant4/collimasim/CMakeLists.txt +0 -26
  50. xcoll/scattering_routines/geant4/collimasim/README.md +0 -21
  51. xcoll/scattering_routines/geant4/collimasim/docs/Makefile +0 -20
  52. xcoll/scattering_routines/geant4/collimasim/docs/make.bat +0 -35
  53. xcoll/scattering_routines/geant4/collimasim/docs/source/collimasim.rst +0 -10
  54. xcoll/scattering_routines/geant4/collimasim/docs/source/conf.py +0 -59
  55. xcoll/scattering_routines/geant4/collimasim/docs/source/index.rst +0 -26
  56. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.appveyor.yml +0 -37
  57. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-format +0 -19
  58. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-tidy +0 -65
  59. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.cmake-format.yaml +0 -73
  60. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.git +0 -1
  61. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CODEOWNERS +0 -9
  62. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CONTRIBUTING.md +0 -386
  63. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/bug-report.yml +0 -45
  64. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/config.yml +0 -8
  65. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/dependabot.yml +0 -16
  66. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler.yml +0 -8
  67. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler_merged.yml +0 -3
  68. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/pull_request_template.md +0 -19
  69. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/ci.yml +0 -969
  70. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/configure.yml +0 -84
  71. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/format.yml +0 -48
  72. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/labeler.yml +0 -16
  73. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/pip.yml +0 -103
  74. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.gitignore +0 -45
  75. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.pre-commit-config.yaml +0 -151
  76. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.readthedocs.yml +0 -3
  77. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/CMakeLists.txt +0 -297
  78. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/LICENSE +0 -29
  79. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/MANIFEST.in +0 -6
  80. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/README.rst +0 -180
  81. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Doxyfile +0 -23
  82. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Makefile +0 -192
  83. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/_static/theme_overrides.css +0 -11
  84. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/chrono.rst +0 -81
  85. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/custom.rst +0 -93
  86. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/eigen.rst +0 -310
  87. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/functional.rst +0 -109
  88. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/index.rst +0 -43
  89. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/overview.rst +0 -171
  90. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/stl.rst +0 -251
  91. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/strings.rst +0 -305
  92. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/classes.rst +0 -1297
  93. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/embedding.rst +0 -262
  94. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/exceptions.rst +0 -396
  95. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/functions.rst +0 -568
  96. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/misc.rst +0 -337
  97. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/index.rst +0 -13
  98. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/numpy.rst +0 -463
  99. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/object.rst +0 -286
  100. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/utilities.rst +0 -155
  101. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/smart_ptrs.rst +0 -174
  102. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/basics.rst +0 -308
  103. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.py +0 -91
  104. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.rst +0 -95
  105. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/changelog.rst +0 -2050
  106. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/classes.rst +0 -542
  107. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/cmake/index.rst +0 -8
  108. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/compiling.rst +0 -648
  109. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/conf.py +0 -381
  110. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/faq.rst +0 -343
  111. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/index.rst +0 -48
  112. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/installing.rst +0 -105
  113. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/limitations.rst +0 -72
  114. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11-logo.png +0 -0
  115. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.png +0 -0
  116. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.svg +0 -427
  117. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.png +0 -0
  118. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.svg +0 -427
  119. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/reference.rst +0 -130
  120. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/release.rst +0 -96
  121. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/requirements.txt +0 -8
  122. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/upgrade.rst +0 -548
  123. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/attr.h +0 -605
  124. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/buffer_info.h +0 -144
  125. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/cast.h +0 -1432
  126. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/chrono.h +0 -213
  127. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/common.h +0 -2
  128. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/complex.h +0 -65
  129. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/class.h +0 -709
  130. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/common.h +0 -1021
  131. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/descr.h +0 -104
  132. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/init.h +0 -346
  133. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/internals.h +0 -467
  134. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/type_caster_base.h +0 -978
  135. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/typeid.h +0 -55
  136. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eigen.h +0 -606
  137. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/embed.h +0 -284
  138. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eval.h +0 -163
  139. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/functional.h +0 -121
  140. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/gil.h +0 -193
  141. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/iostream.h +0 -275
  142. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/numpy.h +0 -1741
  143. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/operators.h +0 -163
  144. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/options.h +0 -65
  145. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pybind11.h +0 -2497
  146. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pytypes.h +0 -1879
  147. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl/filesystem.h +0 -103
  148. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl.h +0 -375
  149. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl_bind.h +0 -747
  150. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/noxfile.py +0 -88
  151. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__init__.py +0 -11
  152. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__main__.py +0 -52
  153. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.py +0 -12
  154. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.pyi +0 -6
  155. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/commands.py +0 -21
  156. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/py.typed +0 -0
  157. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.py +0 -482
  158. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.pyi +0 -63
  159. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pyproject.toml +0 -41
  160. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.cfg +0 -56
  161. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.py +0 -155
  162. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/CMakeLists.txt +0 -503
  163. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/conftest.py +0 -208
  164. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/constructor_stats.h +0 -275
  165. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/cross_module_gil_utils.cpp +0 -73
  166. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/env.py +0 -33
  167. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/pytest.ini +0 -0
  168. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/test_files.py +0 -279
  169. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/pytest.ini +0 -0
  170. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/test_setuphelper.py +0 -143
  171. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/local_bindings.h +0 -85
  172. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/object.h +0 -179
  173. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_cross_module_tests.cpp +0 -151
  174. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.cpp +0 -91
  175. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.h +0 -85
  176. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pytest.ini +0 -19
  177. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/requirements.txt +0 -12
  178. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.cpp +0 -26
  179. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.py +0 -25
  180. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.cpp +0 -216
  181. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.py +0 -163
  182. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.cpp +0 -286
  183. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.py +0 -536
  184. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.cpp +0 -107
  185. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.py +0 -248
  186. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.cpp +0 -227
  187. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.py +0 -202
  188. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.cpp +0 -84
  189. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.py +0 -210
  190. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.cpp +0 -550
  191. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.py +0 -473
  192. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/CMakeLists.txt +0 -84
  193. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/embed.cpp +0 -21
  194. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_embed/CMakeLists.txt +0 -28
  195. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_function/CMakeLists.txt +0 -39
  196. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_target/CMakeLists.txt +0 -46
  197. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/main.cpp +0 -6
  198. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_embed/CMakeLists.txt +0 -41
  199. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_function/CMakeLists.txt +0 -35
  200. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_target/CMakeLists.txt +0 -41
  201. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/test.py +0 -10
  202. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.cpp +0 -165
  203. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.py +0 -53
  204. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.cpp +0 -238
  205. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.py +0 -126
  206. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.cpp +0 -141
  207. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.py +0 -117
  208. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.cpp +0 -41
  209. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.py +0 -50
  210. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.cpp +0 -69
  211. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.py +0 -42
  212. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.cpp +0 -348
  213. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.py +0 -771
  214. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/CMakeLists.txt +0 -47
  215. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/catch.cpp +0 -22
  216. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/external_module.cpp +0 -23
  217. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.cpp +0 -326
  218. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.py +0 -15
  219. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.cpp +0 -148
  220. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.py +0 -272
  221. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.cpp +0 -119
  222. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.py +0 -51
  223. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval_call.py +0 -5
  224. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.cpp +0 -285
  225. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.h +0 -12
  226. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.py +0 -265
  227. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.cpp +0 -397
  228. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.py +0 -520
  229. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.cpp +0 -49
  230. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.py +0 -94
  231. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.cpp +0 -125
  232. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.py +0 -331
  233. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.cpp +0 -153
  234. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.py +0 -284
  235. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.cpp +0 -107
  236. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.py +0 -257
  237. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.cpp +0 -412
  238. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.py +0 -517
  239. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.cpp +0 -102
  240. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.py +0 -92
  241. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.cpp +0 -233
  242. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.py +0 -360
  243. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.cpp +0 -472
  244. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.py +0 -593
  245. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.cpp +0 -524
  246. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.py +0 -441
  247. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.cpp +0 -103
  248. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.py +0 -267
  249. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.cpp +0 -73
  250. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.py +0 -59
  251. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.cpp +0 -235
  252. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.py +0 -146
  253. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.cpp +0 -189
  254. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.py +0 -82
  255. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.cpp +0 -560
  256. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.py +0 -651
  257. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.cpp +0 -500
  258. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.py +0 -253
  259. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.cpp +0 -452
  260. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.py +0 -318
  261. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.cpp +0 -342
  262. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.py +0 -291
  263. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.cpp +0 -131
  264. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.py +0 -318
  265. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.cpp +0 -144
  266. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.py +0 -29
  267. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.cpp +0 -66
  268. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.py +0 -44
  269. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.cpp +0 -22
  270. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.py +0 -9
  271. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.cpp +0 -510
  272. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.py +0 -408
  273. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-numpy-scipy.supp +0 -140
  274. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-python.supp +0 -117
  275. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindCatch.cmake +0 -70
  276. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindEigen3.cmake +0 -86
  277. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindPythonLibsNew.cmake +0 -257
  278. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/check-style.sh +0 -44
  279. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/cmake_uninstall.cmake.in +0 -23
  280. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/libsize.py +0 -39
  281. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/make_changelog.py +0 -64
  282. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Common.cmake +0 -402
  283. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Config.cmake.in +0 -233
  284. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11NewTools.cmake +0 -276
  285. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Tools.cmake +0 -214
  286. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pyproject.toml +0 -3
  287. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_global.py.in +0 -65
  288. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_main.py.in +0 -41
  289. xcoll/scattering_routines/geant4/collimasim/pyproject.toml +0 -8
  290. xcoll/scattering_routines/geant4/collimasim/setup.py +0 -144
  291. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.cpp +0 -403
  292. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.hh +0 -100
  293. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.cpp +0 -763
  294. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.hh +0 -118
  295. xcoll/scattering_routines/geant4/collimasim/src/collimasim/__init__.py +0 -8
  296. xcoll/scattering_routines/geant4/collimasim/src/collimasim/bindings.cpp +0 -63
  297. xcoll/scattering_routines/geant4/collimasim/src/collimasim/pyCollimatorPass.py +0 -142
  298. xcoll/scattering_routines/geant4/collimasim/src/collimasim/xtrack_collimator.py +0 -556
  299. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/PKG-INFO +0 -6
  300. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/SOURCES.txt +0 -24
  301. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/dependency_links.txt +0 -1
  302. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/not-zip-safe +0 -1
  303. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/top_level.txt +0 -1
  304. xcoll/scattering_routines/geant4/collimasim/tests/README.md +0 -25
  305. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_forions.dat +0 -25
  306. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_new_example.dat +0 -18
  307. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_old_example.dat +0 -68
  308. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_testing.dat +0 -15
  309. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_yaml_example.yaml +0 -110
  310. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps.dat +0 -7
  311. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps_pyat_test.dat +0 -3
  312. xcoll/scattering_routines/geant4/collimasim/tests/resources/collonly_twiss_file_example.tfs +0 -54
  313. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings.gmad +0 -3
  314. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_black_absorber.gmad +0 -3
  315. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_ions.gmad +0 -5
  316. xcoll/scattering_routines/geant4/collimasim/tests/resources/twiss_file_testing.tfs +0 -51
  317. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat.py +0 -65
  318. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_passmethod.py +0 -59
  319. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_tracking.py +0 -102
  320. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack.py +0 -75
  321. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_angle.py +0 -74
  322. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_colldb_load.py +0 -84
  323. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction.py +0 -159
  324. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction_ion.py +0 -99
  325. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_ions.py +0 -78
  326. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_lost_energy.py +0 -88
  327. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tilt.py +0 -80
  328. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking.py +0 -97
  329. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking_ions.py +0 -96
  330. xcoll-0.5.12.dist-info/RECORD +0 -415
  331. {xcoll-0.5.12.dist-info → xcoll-0.6.0.dist-info}/LICENSE +0 -0
  332. {xcoll-0.5.12.dist-info → xcoll-0.6.0.dist-info}/NOTICE +0 -0
  333. {xcoll-0.5.12.dist-info → xcoll-0.6.0.dist-info}/WHEEL +0 -0
@@ -1,318 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- import pytest
3
-
4
- import env
5
- from pybind11_tests import stl_binders as m
6
-
7
-
8
- def test_vector_int():
9
- v_int = m.VectorInt([0, 0])
10
- assert len(v_int) == 2
11
- assert bool(v_int) is True
12
-
13
- # test construction from a generator
14
- v_int1 = m.VectorInt(x for x in range(5))
15
- assert v_int1 == m.VectorInt([0, 1, 2, 3, 4])
16
-
17
- v_int2 = m.VectorInt([0, 0])
18
- assert v_int == v_int2
19
- v_int2[1] = 1
20
- assert v_int != v_int2
21
-
22
- v_int2.append(2)
23
- v_int2.insert(0, 1)
24
- v_int2.insert(0, 2)
25
- v_int2.insert(0, 3)
26
- v_int2.insert(6, 3)
27
- assert str(v_int2) == "VectorInt[3, 2, 1, 0, 1, 2, 3]"
28
- with pytest.raises(IndexError):
29
- v_int2.insert(8, 4)
30
-
31
- v_int.append(99)
32
- v_int2[2:-2] = v_int
33
- assert v_int2 == m.VectorInt([3, 2, 0, 0, 99, 2, 3])
34
- del v_int2[1:3]
35
- assert v_int2 == m.VectorInt([3, 0, 99, 2, 3])
36
- del v_int2[0]
37
- assert v_int2 == m.VectorInt([0, 99, 2, 3])
38
-
39
- v_int2.extend(m.VectorInt([4, 5]))
40
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5])
41
-
42
- v_int2.extend([6, 7])
43
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
44
-
45
- # test error handling, and that the vector is unchanged
46
- with pytest.raises(RuntimeError):
47
- v_int2.extend([8, "a"])
48
-
49
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7])
50
-
51
- # test extending from a generator
52
- v_int2.extend(x for x in range(5))
53
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4])
54
-
55
- # test negative indexing
56
- assert v_int2[-1] == 4
57
-
58
- # insert with negative index
59
- v_int2.insert(-1, 88)
60
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88, 4])
61
-
62
- # delete negative index
63
- del v_int2[-1]
64
- assert v_int2 == m.VectorInt([0, 99, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 88])
65
-
66
- v_int2.clear()
67
- assert len(v_int2) == 0
68
-
69
-
70
- # Older PyPy's failed here, related to the PyPy's buffer protocol.
71
- def test_vector_buffer():
72
- b = bytearray([1, 2, 3, 4])
73
- v = m.VectorUChar(b)
74
- assert v[1] == 2
75
- v[2] = 5
76
- mv = memoryview(v) # We expose the buffer interface
77
- if not env.PY2:
78
- assert mv[2] == 5
79
- mv[2] = 6
80
- else:
81
- assert mv[2] == "\x05"
82
- mv[2] = "\x06"
83
- assert v[2] == 6
84
-
85
- if not env.PY2:
86
- mv = memoryview(b)
87
- v = m.VectorUChar(mv[::2])
88
- assert v[1] == 3
89
-
90
- with pytest.raises(RuntimeError) as excinfo:
91
- m.create_undeclstruct() # Undeclared struct contents, no buffer interface
92
- assert "NumPy type info missing for " in str(excinfo.value)
93
-
94
-
95
- def test_vector_buffer_numpy():
96
- np = pytest.importorskip("numpy")
97
- a = np.array([1, 2, 3, 4], dtype=np.int32)
98
- with pytest.raises(TypeError):
99
- m.VectorInt(a)
100
-
101
- a = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]], dtype=np.uintc)
102
- v = m.VectorInt(a[0, :])
103
- assert len(v) == 4
104
- assert v[2] == 3
105
- ma = np.asarray(v)
106
- ma[2] = 5
107
- assert v[2] == 5
108
-
109
- v = m.VectorInt(a[:, 1])
110
- assert len(v) == 3
111
- assert v[2] == 10
112
-
113
- v = m.get_vectorstruct()
114
- assert v[0].x == 5
115
- ma = np.asarray(v)
116
- ma[1]["x"] = 99
117
- assert v[1].x == 99
118
-
119
- v = m.VectorStruct(
120
- np.zeros(
121
- 3,
122
- dtype=np.dtype(
123
- [("w", "bool"), ("x", "I"), ("y", "float64"), ("z", "bool")], align=True
124
- ),
125
- )
126
- )
127
- assert len(v) == 3
128
-
129
- b = np.array([1, 2, 3, 4], dtype=np.uint8)
130
- v = m.VectorUChar(b[::2])
131
- assert v[1] == 3
132
-
133
-
134
- def test_vector_bool():
135
- import pybind11_cross_module_tests as cm
136
-
137
- vv_c = cm.VectorBool()
138
- for i in range(10):
139
- vv_c.append(i % 2 == 0)
140
- for i in range(10):
141
- assert vv_c[i] == (i % 2 == 0)
142
- assert str(vv_c) == "VectorBool[1, 0, 1, 0, 1, 0, 1, 0, 1, 0]"
143
-
144
-
145
- def test_vector_custom():
146
- v_a = m.VectorEl()
147
- v_a.append(m.El(1))
148
- v_a.append(m.El(2))
149
- assert str(v_a) == "VectorEl[El{1}, El{2}]"
150
-
151
- vv_a = m.VectorVectorEl()
152
- vv_a.append(v_a)
153
- vv_b = vv_a[0]
154
- assert str(vv_b) == "VectorEl[El{1}, El{2}]"
155
-
156
-
157
- def test_map_string_double():
158
- mm = m.MapStringDouble()
159
- mm["a"] = 1
160
- mm["b"] = 2.5
161
-
162
- assert list(mm) == ["a", "b"]
163
- assert str(mm) == "MapStringDouble{a: 1, b: 2.5}"
164
- assert "b" in mm
165
- assert "c" not in mm
166
- assert 123 not in mm
167
-
168
- # Check that keys, values, items are views, not merely iterable
169
- keys = mm.keys()
170
- values = mm.values()
171
- items = mm.items()
172
- assert list(keys) == ["a", "b"]
173
- assert len(keys) == 2
174
- assert "a" in keys
175
- assert "c" not in keys
176
- assert 123 not in keys
177
- assert list(items) == [("a", 1), ("b", 2.5)]
178
- assert len(items) == 2
179
- assert ("b", 2.5) in items
180
- assert "hello" not in items
181
- assert ("b", 2.5, None) not in items
182
- assert list(values) == [1, 2.5]
183
- assert len(values) == 2
184
- assert 1 in values
185
- assert 2 not in values
186
- # Check that views update when the map is updated
187
- mm["c"] = -1
188
- assert list(keys) == ["a", "b", "c"]
189
- assert list(values) == [1, 2.5, -1]
190
- assert list(items) == [("a", 1), ("b", 2.5), ("c", -1)]
191
-
192
- um = m.UnorderedMapStringDouble()
193
- um["ua"] = 1.1
194
- um["ub"] = 2.6
195
-
196
- assert sorted(list(um)) == ["ua", "ub"]
197
- assert list(um.keys()) == list(um)
198
- assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
199
- assert list(zip(um.keys(), um.values())) == list(um.items())
200
- assert "UnorderedMapStringDouble" in str(um)
201
-
202
-
203
- def test_map_string_double_const():
204
- mc = m.MapStringDoubleConst()
205
- mc["a"] = 10
206
- mc["b"] = 20.5
207
- assert str(mc) == "MapStringDoubleConst{a: 10, b: 20.5}"
208
-
209
- umc = m.UnorderedMapStringDoubleConst()
210
- umc["a"] = 11
211
- umc["b"] = 21.5
212
-
213
- str(umc)
214
-
215
-
216
- def test_noncopyable_containers():
217
- # std::vector
218
- vnc = m.get_vnc(5)
219
- for i in range(0, 5):
220
- assert vnc[i].value == i + 1
221
-
222
- for i, j in enumerate(vnc, start=1):
223
- assert j.value == i
224
-
225
- # std::deque
226
- dnc = m.get_dnc(5)
227
- for i in range(0, 5):
228
- assert dnc[i].value == i + 1
229
-
230
- i = 1
231
- for j in dnc:
232
- assert j.value == i
233
- i += 1
234
-
235
- # std::map
236
- mnc = m.get_mnc(5)
237
- for i in range(1, 6):
238
- assert mnc[i].value == 10 * i
239
-
240
- vsum = 0
241
- for k, v in mnc.items():
242
- assert v.value == 10 * k
243
- vsum += v.value
244
-
245
- assert vsum == 150
246
-
247
- # std::unordered_map
248
- mnc = m.get_umnc(5)
249
- for i in range(1, 6):
250
- assert mnc[i].value == 10 * i
251
-
252
- vsum = 0
253
- for k, v in mnc.items():
254
- assert v.value == 10 * k
255
- vsum += v.value
256
-
257
- assert vsum == 150
258
-
259
- # nested std::map<std::vector>
260
- nvnc = m.get_nvnc(5)
261
- for i in range(1, 6):
262
- for j in range(0, 5):
263
- assert nvnc[i][j].value == j + 1
264
-
265
- # Note: maps do not have .values()
266
- for _, v in nvnc.items():
267
- for i, j in enumerate(v, start=1):
268
- assert j.value == i
269
-
270
- # nested std::map<std::map>
271
- nmnc = m.get_nmnc(5)
272
- for i in range(1, 6):
273
- for j in range(10, 60, 10):
274
- assert nmnc[i][j].value == 10 * j
275
-
276
- vsum = 0
277
- for _, v_o in nmnc.items():
278
- for k_i, v_i in v_o.items():
279
- assert v_i.value == 10 * k_i
280
- vsum += v_i.value
281
-
282
- assert vsum == 7500
283
-
284
- # nested std::unordered_map<std::unordered_map>
285
- numnc = m.get_numnc(5)
286
- for i in range(1, 6):
287
- for j in range(10, 60, 10):
288
- assert numnc[i][j].value == 10 * j
289
-
290
- vsum = 0
291
- for _, v_o in numnc.items():
292
- for k_i, v_i in v_o.items():
293
- assert v_i.value == 10 * k_i
294
- vsum += v_i.value
295
-
296
- assert vsum == 7500
297
-
298
-
299
- def test_map_delitem():
300
- mm = m.MapStringDouble()
301
- mm["a"] = 1
302
- mm["b"] = 2.5
303
-
304
- assert list(mm) == ["a", "b"]
305
- assert list(mm.items()) == [("a", 1), ("b", 2.5)]
306
- del mm["a"]
307
- assert list(mm) == ["b"]
308
- assert list(mm.items()) == [("b", 2.5)]
309
-
310
- um = m.UnorderedMapStringDouble()
311
- um["ua"] = 1.1
312
- um["ub"] = 2.6
313
-
314
- assert sorted(list(um)) == ["ua", "ub"]
315
- assert sorted(list(um.items())) == [("ua", 1.1), ("ub", 2.6)]
316
- del um["ua"]
317
- assert sorted(list(um)) == ["ub"]
318
- assert sorted(list(um.items())) == [("ub", 2.6)]
@@ -1,144 +0,0 @@
1
- /*
2
- tests/test_tagbased_polymorphic.cpp -- test of polymorphic_type_hook
3
-
4
- Copyright (c) 2018 Hudson River Trading LLC <opensource@hudson-trading.com>
5
-
6
- All rights reserved. Use of this source code is governed by a
7
- BSD-style license that can be found in the LICENSE file.
8
- */
9
-
10
- #include "pybind11_tests.h"
11
- #include <pybind11/stl.h>
12
-
13
- struct Animal
14
- {
15
- // Make this type also a "standard" polymorphic type, to confirm that
16
- // specializing polymorphic_type_hook using enable_if_t still works
17
- // (https://github.com/pybind/pybind11/pull/2016/).
18
- virtual ~Animal() = default;
19
-
20
- // Enum for tag-based polymorphism.
21
- enum class Kind {
22
- Unknown = 0,
23
- Dog = 100, Labrador, Chihuahua, LastDog = 199,
24
- Cat = 200, Panther, LastCat = 299
25
- };
26
- static const std::type_info* type_of_kind(Kind kind);
27
- static std::string name_of_kind(Kind kind);
28
-
29
- const Kind kind;
30
- const std::string name;
31
-
32
- protected:
33
- Animal(const std::string& _name, Kind _kind)
34
- : kind(_kind), name(_name)
35
- {}
36
- };
37
-
38
- struct Dog : Animal
39
- {
40
- explicit Dog(const std::string &_name, Kind _kind = Kind::Dog) : Animal(_name, _kind) {}
41
- std::string bark() const { return name_of_kind(kind) + " " + name + " goes " + sound; }
42
- std::string sound = "WOOF!";
43
- };
44
-
45
- struct Labrador : Dog
46
- {
47
- explicit Labrador(const std::string &_name, int _excitement = 9001)
48
- : Dog(_name, Kind::Labrador), excitement(_excitement) {}
49
- int excitement;
50
- };
51
-
52
- struct Chihuahua : Dog
53
- {
54
- explicit Chihuahua(const std::string &_name) : Dog(_name, Kind::Chihuahua) {
55
- sound = "iyiyiyiyiyi";
56
- }
57
- std::string bark() const { return Dog::bark() + " and runs in circles"; }
58
- };
59
-
60
- struct Cat : Animal
61
- {
62
- explicit Cat(const std::string &_name, Kind _kind = Kind::Cat) : Animal(_name, _kind) {}
63
- std::string purr() const { return "mrowr"; }
64
- };
65
-
66
- struct Panther : Cat
67
- {
68
- explicit Panther(const std::string &_name) : Cat(_name, Kind::Panther) {}
69
- std::string purr() const { return "mrrrRRRRRR"; }
70
- };
71
-
72
- std::vector<std::unique_ptr<Animal>> create_zoo()
73
- {
74
- std::vector<std::unique_ptr<Animal>> ret;
75
- ret.emplace_back(new Labrador("Fido", 15000));
76
-
77
- // simulate some new type of Dog that the Python bindings
78
- // haven't been updated for; it should still be considered
79
- // a Dog, not just an Animal.
80
- ret.emplace_back(new Dog("Ginger", Dog::Kind(150)));
81
-
82
- ret.emplace_back(new Chihuahua("Hertzl"));
83
- ret.emplace_back(new Cat("Tiger", Cat::Kind::Cat));
84
- ret.emplace_back(new Panther("Leo"));
85
- return ret;
86
- }
87
-
88
- const std::type_info* Animal::type_of_kind(Kind kind)
89
- {
90
- switch (kind) {
91
- case Kind::Unknown:
92
- case Kind::Dog: break;
93
-
94
- case Kind::Labrador: return &typeid(Labrador);
95
- case Kind::Chihuahua: return &typeid(Chihuahua);
96
-
97
- case Kind::LastDog:
98
- case Kind::Cat: break;
99
- case Kind::Panther: return &typeid(Panther);
100
- case Kind::LastCat: break;
101
- }
102
-
103
- if (kind >= Kind::Dog && kind <= Kind::LastDog) return &typeid(Dog);
104
- if (kind >= Kind::Cat && kind <= Kind::LastCat) return &typeid(Cat);
105
- return nullptr;
106
- }
107
-
108
- std::string Animal::name_of_kind(Kind kind)
109
- {
110
- std::string raw_name = type_of_kind(kind)->name();
111
- py::detail::clean_type_id(raw_name);
112
- return raw_name;
113
- }
114
-
115
- namespace pybind11 {
116
- template <typename itype>
117
- struct polymorphic_type_hook<itype, detail::enable_if_t<std::is_base_of<Animal, itype>::value>>
118
- {
119
- static const void *get(const itype *src, const std::type_info*& type)
120
- { type = src ? Animal::type_of_kind(src->kind) : nullptr; return src; }
121
- };
122
- } // namespace pybind11
123
-
124
- TEST_SUBMODULE(tagbased_polymorphic, m) {
125
- py::class_<Animal>(m, "Animal")
126
- .def_readonly("name", &Animal::name);
127
- py::class_<Dog, Animal>(m, "Dog")
128
- .def(py::init<std::string>())
129
- .def_readwrite("sound", &Dog::sound)
130
- .def("bark", &Dog::bark);
131
- py::class_<Labrador, Dog>(m, "Labrador")
132
- .def(py::init<std::string, int>(), "name"_a, "excitement"_a = 9001)
133
- .def_readwrite("excitement", &Labrador::excitement);
134
- py::class_<Chihuahua, Dog>(m, "Chihuahua")
135
- .def(py::init<std::string>())
136
- .def("bark", &Chihuahua::bark);
137
- py::class_<Cat, Animal>(m, "Cat")
138
- .def(py::init<std::string>())
139
- .def("purr", &Cat::purr);
140
- py::class_<Panther, Cat>(m, "Panther")
141
- .def(py::init<std::string>())
142
- .def("purr", &Panther::purr);
143
- m.def("create_zoo", &create_zoo);
144
- };
@@ -1,29 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- from pybind11_tests import tagbased_polymorphic as m
3
-
4
-
5
- def test_downcast():
6
- zoo = m.create_zoo()
7
- assert [type(animal) for animal in zoo] == [
8
- m.Labrador,
9
- m.Dog,
10
- m.Chihuahua,
11
- m.Cat,
12
- m.Panther,
13
- ]
14
- assert [animal.name for animal in zoo] == [
15
- "Fido",
16
- "Ginger",
17
- "Hertzl",
18
- "Tiger",
19
- "Leo",
20
- ]
21
- zoo[1].sound = "woooooo"
22
- assert [dog.bark() for dog in zoo[:3]] == [
23
- "Labrador Fido goes WOOF!",
24
- "Dog Ginger goes woooooo",
25
- "Chihuahua Hertzl goes iyiyiyiyiyi and runs in circles",
26
- ]
27
- assert [cat.purr() for cat in zoo[3:]] == ["mrowr", "mrrrRRRRRR"]
28
- zoo[0].excitement -= 1000
29
- assert zoo[0].excitement == 14000
@@ -1,66 +0,0 @@
1
- /*
2
- tests/test_thread.cpp -- call pybind11 bound methods in threads
3
-
4
- Copyright (c) 2021 Laramie Leavitt (Google LLC) <lar@google.com>
5
-
6
- All rights reserved. Use of this source code is governed by a
7
- BSD-style license that can be found in the LICENSE file.
8
- */
9
-
10
- #include <pybind11/cast.h>
11
- #include <pybind11/pybind11.h>
12
-
13
- #include <chrono>
14
- #include <thread>
15
-
16
- #include "pybind11_tests.h"
17
-
18
- namespace py = pybind11;
19
-
20
- namespace {
21
-
22
- struct IntStruct {
23
- explicit IntStruct(int v) : value(v) {};
24
- ~IntStruct() { value = -value; }
25
- IntStruct(const IntStruct&) = default;
26
- IntStruct& operator=(const IntStruct&) = default;
27
-
28
- int value;
29
- };
30
-
31
- } // namespace
32
-
33
- TEST_SUBMODULE(thread, m) {
34
-
35
- py::class_<IntStruct>(m, "IntStruct").def(py::init([](const int i) { return IntStruct(i); }));
36
-
37
- // implicitly_convertible uses loader_life_support when an implicit
38
- // conversion is required in order to lifetime extend the reference.
39
- //
40
- // This test should be run with ASAN for better effectiveness.
41
- py::implicitly_convertible<int, IntStruct>();
42
-
43
- m.def("test", [](int expected, const IntStruct &in) {
44
- {
45
- py::gil_scoped_release release;
46
- std::this_thread::sleep_for(std::chrono::milliseconds(5));
47
- }
48
-
49
- if (in.value != expected) {
50
- throw std::runtime_error("Value changed!!");
51
- }
52
- });
53
-
54
- m.def(
55
- "test_no_gil",
56
- [](int expected, const IntStruct &in) {
57
- std::this_thread::sleep_for(std::chrono::milliseconds(5));
58
- if (in.value != expected) {
59
- throw std::runtime_error("Value changed!!");
60
- }
61
- },
62
- py::call_guard<py::gil_scoped_release>());
63
-
64
- // NOTE: std::string_view also uses loader_life_support to ensure that
65
- // the string contents remain alive, but that's a C++ 17 feature.
66
- }
@@ -1,44 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
-
3
- import threading
4
-
5
- from pybind11_tests import thread as m
6
-
7
-
8
- class Thread(threading.Thread):
9
- def __init__(self, fn):
10
- super(Thread, self).__init__()
11
- self.fn = fn
12
- self.e = None
13
-
14
- def run(self):
15
- try:
16
- for i in range(10):
17
- self.fn(i, i)
18
- except Exception as e:
19
- self.e = e
20
-
21
- def join(self):
22
- super(Thread, self).join()
23
- if self.e:
24
- raise self.e
25
-
26
-
27
- def test_implicit_conversion():
28
- a = Thread(m.test)
29
- b = Thread(m.test)
30
- c = Thread(m.test)
31
- for x in [a, b, c]:
32
- x.start()
33
- for x in [c, b, a]:
34
- x.join()
35
-
36
-
37
- def test_implicit_conversion_no_gil():
38
- a = Thread(m.test_no_gil)
39
- b = Thread(m.test_no_gil)
40
- c = Thread(m.test_no_gil)
41
- for x in [a, b, c]:
42
- x.start()
43
- for x in [c, b, a]:
44
- x.join()
@@ -1,22 +0,0 @@
1
- /*
2
- tests/test_class.cpp -- test py::class_ definitions and basic functionality
3
-
4
- Copyright (c) 2019 Roland Dreier <roland.dreier@gmail.com>
5
-
6
- All rights reserved. Use of this source code is governed by a
7
- BSD-style license that can be found in the LICENSE file.
8
- */
9
-
10
- #include "pybind11_tests.h"
11
-
12
- TEST_SUBMODULE(union_, m) {
13
- union TestUnion {
14
- int value_int;
15
- unsigned value_uint;
16
- };
17
-
18
- py::class_<TestUnion>(m, "TestUnion")
19
- .def(py::init<>())
20
- .def_readonly("as_int", &TestUnion::value_int)
21
- .def_readwrite("as_uint", &TestUnion::value_uint);
22
- }
@@ -1,9 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- from pybind11_tests import union_ as m
3
-
4
-
5
- def test_union():
6
- instance = m.TestUnion()
7
-
8
- instance.as_uint = 10
9
- assert instance.as_int == 10