123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- # Get Python six functionality:
- from __future__ import\
- absolute_import, print_function, division, unicode_literals
- ###############################################################################
- ###############################################################################
- ###############################################################################
- import pytest
- from innvestigate.utils.tests import dryrun
- from innvestigate.analyzer import Input
- from innvestigate.analyzer import Random
- ###############################################################################
- ###############################################################################
- ###############################################################################
- @pytest.mark.fast
- @pytest.mark.precommit
- def test_fast__Input():
- def method(model):
- return Input(model)
- dryrun.test_analyzer(method, "trivia.*:mnist.log_reg")
- @pytest.mark.fast
- @pytest.mark.precommit
- def test_fast__Random():
- def method(model):
- return Random(model)
- dryrun.test_analyzer(method, "trivia.*:mnist.log_reg")
- ###############################################################################
- ###############################################################################
- ###############################################################################
- @pytest.mark.fast
- @pytest.mark.precommit
- def test_fast__SerializeRandom():
- def method(model):
- return Random(model)
- dryrun.test_serialize_analyzer(method, "trivia.*:mnist.log_reg")
|