This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description
This PR refactors ONNX import/export test code
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Current directory structure:
Redundant code:
Changes that are part of cleanup:
Create onnx/mxnet_backend_test.py and onnx/gluon_backend_test.py for import/export with MXNet/Gluon backend respectively. These can't be combined further because the ONNX tests are added in globals() dictionary with test name as key. Combining these 2 will override the test list in globals().
Reference this newly created test_cases.py in mxnet_backend_test.py, onnx_backend_test.py, gluon_backend_test.py. Create a common onnx/backend_test.py for preparing the test list.
Example call:
Option 1:
Option 2:
To add tests for models, add a line in test_cases[]
Example:
To add tests for operators,
Directory structure after cleanup:
Future work:
Comments