| from __future__ import absolute_import, division, print_function, unicode_literals | |
| import torch | |
| from common_utils import TestCase, run_tests | |
| class TestFunctionSchema(TestCase): | |
| def test_serialize_and_deserialize(self): | |
| schemas = torch._C._jit_get_all_schemas() | |
| # so far we have around 1700 registered schemas | |
| self.assertGreater(len(schemas), 1000) | |
| for schema in schemas: | |
| parsed_schema = torch._C.parse_schema(str(schema)) | |
| self.assertEqual(parsed_schema, schema) | |
| if __name__ == '__main__': | |
| run_tests() |