/usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/tests
NameSizeModeActions
__pycache__/-0755rm
c2_ref_test.py315770644editdlrm
conversion_test.py151590644editdlrm
helper_test.py8070644editdlrm
onnx_backend_test.py98910644editdlrm
ssa_test.py40160644editdlrm
test_utils.py7870644editdlrm
__init__.py00644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/caffe2/python/onnx/tests/test_utils.py (787B)
## @package onnx # Module caffe2.python.onnx.tests.test_utils import unittest import numpy as np class TestCase(unittest.TestCase): def setUp(self): np.random.seed(seed=0) def assertSameOutputs(self, outputs1, outputs2, decimal=7): self.assertEqual(len(outputs1), len(outputs2)) for o1, o2 in zip(outputs1, outputs2): self.assertEqual(o1.dtype, o2.dtype) np.testing.assert_almost_equal(o1, o2, decimal=decimal) def add_test_case(self, name, test_func): if not name.startswith('test_'): raise ValueError('Test name must start with test_: {}'.format(name)) if hasattr(self, name): raise ValueError('Duplicated test name: {}'.format(name)) setattr(self, name, test_func)