test_basic.py 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. import builtins
  2. import os
  3. import subprocess
  4. import sys
  5. import textwrap
  6. def test_simple():
  7. assert 1 + 1 == 2
  8. def test_override_builtins():
  9. import pylab
  10. ok_to_override = {
  11. '__name__',
  12. '__doc__',
  13. '__package__',
  14. '__loader__',
  15. '__spec__',
  16. 'any',
  17. 'all',
  18. 'sum',
  19. 'divmod'
  20. }
  21. overridden = False
  22. for key in dir(pylab):
  23. if key in dir(builtins):
  24. if (getattr(pylab, key) != getattr(builtins, key) and
  25. key not in ok_to_override):
  26. print("'%s' was overridden in globals()." % key)
  27. overridden = True
  28. assert not overridden
  29. def test_lazy_imports():
  30. source = textwrap.dedent("""
  31. import sys
  32. import matplotlib.figure
  33. import matplotlib.backend_bases
  34. import matplotlib.pyplot
  35. assert 'matplotlib._tri' not in sys.modules
  36. assert 'matplotlib._qhull' not in sys.modules
  37. assert 'matplotlib._contour' not in sys.modules
  38. assert 'urllib.request' not in sys.modules
  39. """)
  40. subprocess.check_call(
  41. [sys.executable, '-c', source],
  42. env={**os.environ, "MPLBACKEND": "", "MATPLOTLIBRC": os.devnull})