test_modules.py 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. # -*- coding: utf-8 -*-
  2. from pybind11_tests import modules as m
  3. from pybind11_tests.modules import subsubmodule as ms
  4. from pybind11_tests import ConstructorStats
  5. def test_nested_modules():
  6. import pybind11_tests
  7. assert pybind11_tests.__name__ == "pybind11_tests"
  8. assert pybind11_tests.modules.__name__ == "pybind11_tests.modules"
  9. assert (
  10. pybind11_tests.modules.subsubmodule.__name__
  11. == "pybind11_tests.modules.subsubmodule"
  12. )
  13. assert m.__name__ == "pybind11_tests.modules"
  14. assert ms.__name__ == "pybind11_tests.modules.subsubmodule"
  15. assert ms.submodule_func() == "submodule_func()"
  16. def test_reference_internal():
  17. b = ms.B()
  18. assert str(b.get_a1()) == "A[1]"
  19. assert str(b.a1) == "A[1]"
  20. assert str(b.get_a2()) == "A[2]"
  21. assert str(b.a2) == "A[2]"
  22. b.a1 = ms.A(42)
  23. b.a2 = ms.A(43)
  24. assert str(b.get_a1()) == "A[42]"
  25. assert str(b.a1) == "A[42]"
  26. assert str(b.get_a2()) == "A[43]"
  27. assert str(b.a2) == "A[43]"
  28. astats, bstats = ConstructorStats.get(ms.A), ConstructorStats.get(ms.B)
  29. assert astats.alive() == 2
  30. assert bstats.alive() == 1
  31. del b
  32. assert astats.alive() == 0
  33. assert bstats.alive() == 0
  34. assert astats.values() == ["1", "2", "42", "43"]
  35. assert bstats.values() == []
  36. assert astats.default_constructions == 0
  37. assert bstats.default_constructions == 1
  38. assert astats.copy_constructions == 0
  39. assert bstats.copy_constructions == 0
  40. # assert astats.move_constructions >= 0 # Don't invoke any
  41. # assert bstats.move_constructions >= 0 # Don't invoke any
  42. assert astats.copy_assignments == 2
  43. assert bstats.copy_assignments == 0
  44. assert astats.move_assignments == 0
  45. assert bstats.move_assignments == 0
  46. def test_importing():
  47. from pybind11_tests.modules import OD
  48. from collections import OrderedDict
  49. assert OD is OrderedDict
  50. assert str(OD([(1, "a"), (2, "b")])) == "OrderedDict([(1, 'a'), (2, 'b')])"
  51. def test_pydoc():
  52. """Pydoc needs to be able to provide help() for everything inside a pybind11 module"""
  53. import pybind11_tests
  54. import pydoc
  55. assert pybind11_tests.__name__ == "pybind11_tests"
  56. assert pybind11_tests.__doc__ == "pybind11 test module"
  57. assert pydoc.text.docmodule(pybind11_tests)
  58. def test_duplicate_registration():
  59. """Registering two things with the same name"""
  60. assert m.duplicate_registration() == []
  61. def test_builtin_key_type():
  62. """Test that all the keys in the builtin modules have type str.
  63. Previous versions of pybind11 would add a unicode key in python 2.
  64. """
  65. if hasattr(__builtins__, "keys"):
  66. keys = __builtins__.keys()
  67. else: # this is to make pypy happy since builtins is different there.
  68. keys = __builtins__.__dict__.keys()
  69. assert {type(k) for k in keys} == {str}