test_misc.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. # Get Python six functionality:
  2. from __future__ import\
  3. absolute_import, print_function, division, unicode_literals
  4. ###############################################################################
  5. ###############################################################################
  6. ###############################################################################
  7. import pytest
  8. from innvestigate.utils.tests import dryrun
  9. from innvestigate.analyzer import Input
  10. from innvestigate.analyzer import Random
  11. ###############################################################################
  12. ###############################################################################
  13. ###############################################################################
  14. @pytest.mark.fast
  15. @pytest.mark.precommit
  16. def test_fast__Input():
  17. def method(model):
  18. return Input(model)
  19. dryrun.test_analyzer(method, "trivia.*:mnist.log_reg")
  20. @pytest.mark.fast
  21. @pytest.mark.precommit
  22. def test_fast__Random():
  23. def method(model):
  24. return Random(model)
  25. dryrun.test_analyzer(method, "trivia.*:mnist.log_reg")
  26. ###############################################################################
  27. ###############################################################################
  28. ###############################################################################
  29. @pytest.mark.fast
  30. @pytest.mark.precommit
  31. def test_fast__SerializeRandom():
  32. def method(model):
  33. return Random(model)
  34. dryrun.test_serialize_analyzer(method, "trivia.*:mnist.log_reg")