test_show_versions.py 1006 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. from sklearn.utils._show_versions import _get_deps_info, _get_sys_info, show_versions
  2. from sklearn.utils._testing import ignore_warnings
  3. from sklearn.utils.fixes import threadpool_info
  4. def test_get_sys_info():
  5. sys_info = _get_sys_info()
  6. assert "python" in sys_info
  7. assert "executable" in sys_info
  8. assert "machine" in sys_info
  9. def test_get_deps_info():
  10. with ignore_warnings():
  11. deps_info = _get_deps_info()
  12. assert "pip" in deps_info
  13. assert "setuptools" in deps_info
  14. assert "sklearn" in deps_info
  15. assert "numpy" in deps_info
  16. assert "scipy" in deps_info
  17. assert "Cython" in deps_info
  18. assert "pandas" in deps_info
  19. assert "matplotlib" in deps_info
  20. assert "joblib" in deps_info
  21. def test_show_versions(capsys):
  22. with ignore_warnings():
  23. show_versions()
  24. out, err = capsys.readouterr()
  25. assert "python" in out
  26. assert "numpy" in out
  27. info = threadpool_info()
  28. if info:
  29. assert "threadpoolctl info:" in out