old_contents: import unittest
  
  from onnx import defs, AttributeProto
  
  
  class TestSchema(unittest.TestCase):
  
      def test_get_schema(self):
          defs.get_schema("Relu")
  
      def test_typecheck(self):
          defs.get_schema("Conv")
  
      def test_attr_default_value(self):
          v = defs.get_schema(
              "BatchNormalization").attributes['epsilon'].default_value
          self.assertEqual(type(v), AttributeProto)
          self.assertEqual(v.type, AttributeProto.FLOAT)
  
  
  if __name__ == '__main__':
      unittest.main()
  
new_contents: import unittest
  
  from onnx import defs, AttributeProto
  
  
  class TestSchema(unittest.TestCase):
  
      def test_get_schema(self):  # type: () -> None
          defs.get_schema("Relu")
  
      def test_typecheck(self):  # type: () -> None
          defs.get_schema("Conv")
  
      def test_attr_default_value(self):
          v = defs.get_schema(
              "BatchNormalization").attributes['epsilon'].default_value
          self.assertEqual(type(v), AttributeProto)
          self.assertEqual(v.type, AttributeProto.FLOAT)
  
  
  if __name__ == '__main__':
      unittest.main()
  
subject: Add type hints to schema_test.py (#953)