xcoll 0.5.12__py3-none-any.whl → 0.6.1__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 (336) hide show
  1. xcoll/__init__.py +5 -18
  2. xcoll/__init__.py.orig +26 -0
  3. xcoll/beam_elements/__init__.py +1 -0
  4. xcoll/beam_elements/absorber.py +12 -2
  5. xcoll/beam_elements/base.py +162 -62
  6. xcoll/beam_elements/blowup.py +1 -0
  7. xcoll/beam_elements/elements_src/black_absorber.h +57 -58
  8. xcoll/beam_elements/elements_src/black_crystal.h +49 -50
  9. xcoll/beam_elements/elements_src/everest_block.h +21 -11
  10. xcoll/beam_elements/elements_src/everest_collimator.h +100 -91
  11. xcoll/beam_elements/elements_src/everest_crystal.h +145 -140
  12. xcoll/beam_elements/elements_src/transparent_collimator.h +126 -0
  13. xcoll/beam_elements/elements_src/transparent_crystal.h +118 -0
  14. xcoll/beam_elements/everest.py +16 -5
  15. xcoll/beam_elements/monitor.py +1 -0
  16. xcoll/beam_elements/transparent.py +83 -0
  17. xcoll/colldb.py +15 -6
  18. xcoll/general.py +1 -1
  19. xcoll/general.py.orig +20 -0
  20. xcoll/headers/particle_states.py +51 -0
  21. xcoll/initial_distribution.py +121 -91
  22. xcoll/initial_distribution.py.orig +256 -0
  23. xcoll/interaction_record/interaction_record.py +2 -1
  24. xcoll/interaction_record/interaction_types.py +2 -2
  25. xcoll/line_tools.py +163 -73
  26. xcoll/lossmap.py +519 -127
  27. xcoll/plot.py +109 -0
  28. xcoll/rf_sweep.py +6 -0
  29. xcoll/scattering_routines/engine.py +414 -217
  30. xcoll/scattering_routines/environment.py +297 -0
  31. xcoll/scattering_routines/everest/amorphous.h +95 -71
  32. xcoll/scattering_routines/everest/{channeling.h → channelling.h} +121 -112
  33. xcoll/scattering_routines/everest/constants.h +1 -1
  34. xcoll/scattering_routines/everest/crystal_parameters.h +9 -9
  35. xcoll/scattering_routines/everest/everest.h +8 -3
  36. xcoll/scattering_routines/everest/everest.py +2 -1
  37. xcoll/scattering_routines/everest/ionisation_loss.h +141 -0
  38. xcoll/scattering_routines/everest/jaw.h +19 -24
  39. xcoll/scattering_routines/everest/materials.py +2 -0
  40. xcoll/scattering_routines/everest/multiple_coulomb_scattering.h +2 -2
  41. xcoll/scattering_routines/everest/nuclear_interaction.h +35 -19
  42. xcoll/scattering_routines/everest/properties.h +3 -72
  43. xcoll/xaux.py +65 -109
  44. {xcoll-0.5.12.dist-info → xcoll-0.6.1.dist-info}/METADATA +5 -5
  45. xcoll-0.6.1.dist-info/RECORD +138 -0
  46. xcoll/_manager.py +0 -22
  47. xcoll/headers/particle_states.h +0 -25
  48. xcoll/install.py +0 -35
  49. xcoll/scattering_routines/geant4/collimasim/.git +0 -1
  50. xcoll/scattering_routines/geant4/collimasim/.gitignore +0 -12
  51. xcoll/scattering_routines/geant4/collimasim/.gitmodules +0 -3
  52. xcoll/scattering_routines/geant4/collimasim/CMakeLists.txt +0 -26
  53. xcoll/scattering_routines/geant4/collimasim/README.md +0 -21
  54. xcoll/scattering_routines/geant4/collimasim/docs/Makefile +0 -20
  55. xcoll/scattering_routines/geant4/collimasim/docs/make.bat +0 -35
  56. xcoll/scattering_routines/geant4/collimasim/docs/source/collimasim.rst +0 -10
  57. xcoll/scattering_routines/geant4/collimasim/docs/source/conf.py +0 -59
  58. xcoll/scattering_routines/geant4/collimasim/docs/source/index.rst +0 -26
  59. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.appveyor.yml +0 -37
  60. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-format +0 -19
  61. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.clang-tidy +0 -65
  62. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.cmake-format.yaml +0 -73
  63. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.git +0 -1
  64. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CODEOWNERS +0 -9
  65. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/CONTRIBUTING.md +0 -386
  66. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/bug-report.yml +0 -45
  67. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/ISSUE_TEMPLATE/config.yml +0 -8
  68. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/dependabot.yml +0 -16
  69. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler.yml +0 -8
  70. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/labeler_merged.yml +0 -3
  71. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/pull_request_template.md +0 -19
  72. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/ci.yml +0 -969
  73. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/configure.yml +0 -84
  74. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/format.yml +0 -48
  75. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/labeler.yml +0 -16
  76. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.github/workflows/pip.yml +0 -103
  77. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.gitignore +0 -45
  78. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.pre-commit-config.yaml +0 -151
  79. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/.readthedocs.yml +0 -3
  80. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/CMakeLists.txt +0 -297
  81. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/LICENSE +0 -29
  82. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/MANIFEST.in +0 -6
  83. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/README.rst +0 -180
  84. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Doxyfile +0 -23
  85. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/Makefile +0 -192
  86. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/_static/theme_overrides.css +0 -11
  87. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/chrono.rst +0 -81
  88. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/custom.rst +0 -93
  89. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/eigen.rst +0 -310
  90. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/functional.rst +0 -109
  91. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/index.rst +0 -43
  92. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/overview.rst +0 -171
  93. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/stl.rst +0 -251
  94. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/cast/strings.rst +0 -305
  95. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/classes.rst +0 -1297
  96. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/embedding.rst +0 -262
  97. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/exceptions.rst +0 -396
  98. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/functions.rst +0 -568
  99. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/misc.rst +0 -337
  100. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/index.rst +0 -13
  101. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/numpy.rst +0 -463
  102. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/object.rst +0 -286
  103. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/pycpp/utilities.rst +0 -155
  104. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/advanced/smart_ptrs.rst +0 -174
  105. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/basics.rst +0 -308
  106. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.py +0 -91
  107. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/benchmark.rst +0 -95
  108. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/changelog.rst +0 -2050
  109. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/classes.rst +0 -542
  110. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/cmake/index.rst +0 -8
  111. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/compiling.rst +0 -648
  112. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/conf.py +0 -381
  113. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/faq.rst +0 -343
  114. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/index.rst +0 -48
  115. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/installing.rst +0 -105
  116. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/limitations.rst +0 -72
  117. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11-logo.png +0 -0
  118. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.png +0 -0
  119. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python1.svg +0 -427
  120. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.png +0 -0
  121. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/pybind11_vs_boost_python2.svg +0 -427
  122. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/reference.rst +0 -130
  123. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/release.rst +0 -96
  124. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/requirements.txt +0 -8
  125. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/docs/upgrade.rst +0 -548
  126. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/attr.h +0 -605
  127. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/buffer_info.h +0 -144
  128. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/cast.h +0 -1432
  129. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/chrono.h +0 -213
  130. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/common.h +0 -2
  131. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/complex.h +0 -65
  132. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/class.h +0 -709
  133. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/common.h +0 -1021
  134. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/descr.h +0 -104
  135. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/init.h +0 -346
  136. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/internals.h +0 -467
  137. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/type_caster_base.h +0 -978
  138. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/detail/typeid.h +0 -55
  139. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eigen.h +0 -606
  140. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/embed.h +0 -284
  141. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/eval.h +0 -163
  142. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/functional.h +0 -121
  143. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/gil.h +0 -193
  144. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/iostream.h +0 -275
  145. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/numpy.h +0 -1741
  146. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/operators.h +0 -163
  147. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/options.h +0 -65
  148. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pybind11.h +0 -2497
  149. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/pytypes.h +0 -1879
  150. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl/filesystem.h +0 -103
  151. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl.h +0 -375
  152. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/include/pybind11/stl_bind.h +0 -747
  153. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/noxfile.py +0 -88
  154. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__init__.py +0 -11
  155. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/__main__.py +0 -52
  156. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.py +0 -12
  157. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/_version.pyi +0 -6
  158. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/commands.py +0 -21
  159. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/py.typed +0 -0
  160. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.py +0 -482
  161. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pybind11/setup_helpers.pyi +0 -63
  162. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/pyproject.toml +0 -41
  163. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.cfg +0 -56
  164. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/setup.py +0 -155
  165. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/CMakeLists.txt +0 -503
  166. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/conftest.py +0 -208
  167. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/constructor_stats.h +0 -275
  168. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/cross_module_gil_utils.cpp +0 -73
  169. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/env.py +0 -33
  170. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/pytest.ini +0 -0
  171. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_python_package/test_files.py +0 -279
  172. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/pytest.ini +0 -0
  173. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/extra_setuptools/test_setuphelper.py +0 -143
  174. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/local_bindings.h +0 -85
  175. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/object.h +0 -179
  176. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_cross_module_tests.cpp +0 -151
  177. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.cpp +0 -91
  178. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pybind11_tests.h +0 -85
  179. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/pytest.ini +0 -19
  180. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/requirements.txt +0 -12
  181. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.cpp +0 -26
  182. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_async.py +0 -25
  183. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.cpp +0 -216
  184. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_buffers.py +0 -163
  185. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.cpp +0 -286
  186. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_builtin_casters.py +0 -536
  187. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.cpp +0 -107
  188. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_call_policies.py +0 -248
  189. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.cpp +0 -227
  190. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_callbacks.py +0 -202
  191. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.cpp +0 -84
  192. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_chrono.py +0 -210
  193. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.cpp +0 -550
  194. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_class.py +0 -473
  195. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/CMakeLists.txt +0 -84
  196. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/embed.cpp +0 -21
  197. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_embed/CMakeLists.txt +0 -28
  198. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_function/CMakeLists.txt +0 -39
  199. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/installed_target/CMakeLists.txt +0 -46
  200. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/main.cpp +0 -6
  201. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_embed/CMakeLists.txt +0 -41
  202. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_function/CMakeLists.txt +0 -35
  203. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/subdirectory_target/CMakeLists.txt +0 -41
  204. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_cmake_build/test.py +0 -10
  205. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.cpp +0 -165
  206. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_constants_and_functions.py +0 -53
  207. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.cpp +0 -238
  208. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_copy_move.py +0 -126
  209. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.cpp +0 -141
  210. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_casters.py +0 -117
  211. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.cpp +0 -41
  212. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_custom_type_setup.py +0 -50
  213. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.cpp +0 -69
  214. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_docstring_options.py +0 -42
  215. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.cpp +0 -348
  216. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eigen.py +0 -771
  217. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/CMakeLists.txt +0 -47
  218. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/catch.cpp +0 -22
  219. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/external_module.cpp +0 -23
  220. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.cpp +0 -326
  221. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_embed/test_interpreter.py +0 -15
  222. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.cpp +0 -148
  223. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_enum.py +0 -272
  224. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.cpp +0 -119
  225. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval.py +0 -51
  226. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_eval_call.py +0 -5
  227. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.cpp +0 -285
  228. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.h +0 -12
  229. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_exceptions.py +0 -265
  230. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.cpp +0 -397
  231. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_factory_constructors.py +0 -520
  232. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.cpp +0 -49
  233. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_gil_scoped.py +0 -94
  234. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.cpp +0 -125
  235. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_iostream.py +0 -331
  236. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.cpp +0 -153
  237. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_kwargs_and_defaults.py +0 -284
  238. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.cpp +0 -107
  239. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_local_bindings.py +0 -257
  240. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.cpp +0 -412
  241. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_methods_and_attributes.py +0 -517
  242. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.cpp +0 -102
  243. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_modules.py +0 -92
  244. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.cpp +0 -233
  245. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_multiple_inheritance.py +0 -360
  246. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.cpp +0 -472
  247. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_array.py +0 -593
  248. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.cpp +0 -524
  249. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_dtypes.py +0 -441
  250. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.cpp +0 -103
  251. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_numpy_vectorize.py +0 -267
  252. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.cpp +0 -73
  253. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_opaque_types.py +0 -59
  254. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.cpp +0 -235
  255. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_operator_overloading.py +0 -146
  256. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.cpp +0 -189
  257. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pickling.py +0 -82
  258. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.cpp +0 -560
  259. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_pytypes.py +0 -651
  260. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.cpp +0 -500
  261. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_sequences_and_iterators.py +0 -253
  262. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.cpp +0 -452
  263. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_smart_ptr.py +0 -318
  264. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.cpp +0 -342
  265. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl.py +0 -291
  266. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.cpp +0 -131
  267. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_stl_binders.py +0 -318
  268. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.cpp +0 -144
  269. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_tagbased_polymorphic.py +0 -29
  270. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.cpp +0 -66
  271. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_thread.py +0 -44
  272. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.cpp +0 -22
  273. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_union.py +0 -9
  274. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.cpp +0 -510
  275. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/test_virtual_functions.py +0 -408
  276. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-numpy-scipy.supp +0 -140
  277. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tests/valgrind-python.supp +0 -117
  278. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindCatch.cmake +0 -70
  279. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindEigen3.cmake +0 -86
  280. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/FindPythonLibsNew.cmake +0 -257
  281. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/check-style.sh +0 -44
  282. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/cmake_uninstall.cmake.in +0 -23
  283. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/libsize.py +0 -39
  284. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/make_changelog.py +0 -64
  285. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Common.cmake +0 -402
  286. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Config.cmake.in +0 -233
  287. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11NewTools.cmake +0 -276
  288. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pybind11Tools.cmake +0 -214
  289. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/pyproject.toml +0 -3
  290. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_global.py.in +0 -65
  291. xcoll/scattering_routines/geant4/collimasim/lib/pybind11/tools/setup_main.py.in +0 -41
  292. xcoll/scattering_routines/geant4/collimasim/pyproject.toml +0 -8
  293. xcoll/scattering_routines/geant4/collimasim/setup.py +0 -144
  294. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.cpp +0 -403
  295. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSPyATInterface.hh +0 -100
  296. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.cpp +0 -763
  297. xcoll/scattering_routines/geant4/collimasim/src/collimasim/BDSXtrackInterface.hh +0 -118
  298. xcoll/scattering_routines/geant4/collimasim/src/collimasim/__init__.py +0 -8
  299. xcoll/scattering_routines/geant4/collimasim/src/collimasim/bindings.cpp +0 -63
  300. xcoll/scattering_routines/geant4/collimasim/src/collimasim/pyCollimatorPass.py +0 -142
  301. xcoll/scattering_routines/geant4/collimasim/src/collimasim/xtrack_collimator.py +0 -556
  302. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/PKG-INFO +0 -6
  303. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/SOURCES.txt +0 -24
  304. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/dependency_links.txt +0 -1
  305. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/not-zip-safe +0 -1
  306. xcoll/scattering_routines/geant4/collimasim/src/collimasim.egg-info/top_level.txt +0 -1
  307. xcoll/scattering_routines/geant4/collimasim/tests/README.md +0 -25
  308. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_forions.dat +0 -25
  309. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_new_example.dat +0 -18
  310. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_old_example.dat +0 -68
  311. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_testing.dat +0 -15
  312. xcoll/scattering_routines/geant4/collimasim/tests/resources/CollDB_yaml_example.yaml +0 -110
  313. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps.dat +0 -7
  314. xcoll/scattering_routines/geant4/collimasim/tests/resources/collgaps_pyat_test.dat +0 -3
  315. xcoll/scattering_routines/geant4/collimasim/tests/resources/collonly_twiss_file_example.tfs +0 -54
  316. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings.gmad +0 -3
  317. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_black_absorber.gmad +0 -3
  318. xcoll/scattering_routines/geant4/collimasim/tests/resources/settings_ions.gmad +0 -5
  319. xcoll/scattering_routines/geant4/collimasim/tests/resources/twiss_file_testing.tfs +0 -51
  320. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat.py +0 -65
  321. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_passmethod.py +0 -59
  322. xcoll/scattering_routines/geant4/collimasim/tests/test_pyat_tracking.py +0 -102
  323. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack.py +0 -75
  324. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_angle.py +0 -74
  325. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_colldb_load.py +0 -84
  326. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction.py +0 -159
  327. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_interaction_ion.py +0 -99
  328. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_ions.py +0 -78
  329. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_lost_energy.py +0 -88
  330. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tilt.py +0 -80
  331. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking.py +0 -97
  332. xcoll/scattering_routines/geant4/collimasim/tests/test_xtrack_tracking_ions.py +0 -96
  333. xcoll-0.5.12.dist-info/RECORD +0 -415
  334. {xcoll-0.5.12.dist-info → xcoll-0.6.1.dist-info}/LICENSE +0 -0
  335. {xcoll-0.5.12.dist-info → xcoll-0.6.1.dist-info}/NOTICE +0 -0
  336. {xcoll-0.5.12.dist-info → xcoll-0.6.1.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