Skip to content

Commit b1ffafe

Browse files
Move keras2onnx unit tests inside tests dir (#1591)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 106b85b commit b1ffafe

File tree

9 files changed

+6
-3
lines changed

9 files changed

+6
-3
lines changed

ci_build/azure_pipelines/keras2onnx_unit_test.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ jobs:
8585
pip install $(ONNX_PATH)
8686
pip install h5py==2.9.0
8787
pip install numpy==1.19
88+
pip install parameterized
8889
pip install $(TENSORFLOW_PATH)
8990
if [[ ! -z $KERAS ]];
9091
then
@@ -105,7 +106,7 @@ jobs:
105106
then
106107
export TF_KERAS=0
107108
fi
108-
pytest keras2onnx_tests --doctest-modules --junitxml=junit/test-results.xml
109+
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
109110
displayName: 'pytest'
110111
111112
- task: PublishTestResults@2
@@ -198,6 +199,7 @@ jobs:
198199
pip uninstall -y protobuf
199200
pip install protobuf
200201
pip install h5py==2.9.0
202+
pip install parameterized
201203
pip install %TENSORFLOW_PATH%
202204
IF NOT "%KERAS%"=="" (pip install %KERAS%)
203205
pip install git+https://github.com/microsoft/onnxconverter-common
@@ -211,7 +213,7 @@ jobs:
211213
echo Test onnxruntime installation... && python -c "import onnxruntime"
212214
python -c "import onnxconverter_common"
213215
IF NOT "%KERAS%"=="" (set TF_KERAS=0)
214-
pytest keras2onnx_tests --doctest-modules --junitxml=junit/test-results.xml
216+
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
215217
displayName: 'pytest'
216218
217219
- task: PublishTestResults@2

ci_build/azure_pipelines/templates/unit_test.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@ steps:
1212
export TF2ONNX_TEST_OPSET=$CI_ONNX_OPSET
1313
export TF2ONNX_SKIP_TFLITE_TESTS=$CI_SKIP_TFLITE_TESTS
1414
export TF2ONNX_SKIP_TF_TESTS=$CI_SKIP_TF_TESTS
15-
python -m pytest --cov=tf2onnx --cov-report=term --disable-pytest-warnings -r s tests --cov-append
15+
python -m pytest --cov=tf2onnx --cov-report=term --disable-pytest-warnings -r s tests --cov-append \
16+
--ignore tests/keras2onnx_unit_tests
1617
timeoutInMinutes: 15
1718
displayName: ${{ format('Run UnitTest - Opset{0}', onnx_opset) }}
1819
condition: succeededOrFailed()

0 commit comments

Comments
 (0)