Add ability to skip applications test via env var.
This commit is contained in:
parent
e23c55d26d
commit
931e008683
@ -1,3 +1,5 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import pytest
|
import pytest
|
||||||
from absl.testing import parameterized
|
from absl.testing import parameterized
|
||||||
@ -101,6 +103,10 @@ def _get_elephant(target_size):
|
|||||||
return np.expand_dims(x, axis=0)
|
return np.expand_dims(x, axis=0)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.skipif(
|
||||||
|
os.environ.get("SKIP_APPLICATIONS_TESTS"),
|
||||||
|
reason="Env variable set to skip.",
|
||||||
|
)
|
||||||
class ApplicationsTest(testing.TestCase, parameterized.TestCase):
|
class ApplicationsTest(testing.TestCase, parameterized.TestCase):
|
||||||
@parameterized.named_parameters(MODEL_LIST)
|
@parameterized.named_parameters(MODEL_LIST)
|
||||||
def test_application_notop_variable_input_channels(self, app, last_dim, _):
|
def test_application_notop_variable_input_channels(self, app, last_dim, _):
|
||||||
|
Loading…
Reference in New Issue
Block a user