def test_TypeContract_Set_Union(self):
theirs_tc = pydoc.DocString(func_set_union).get_type_contract()
ours_tc = pydoc.TypeContract(["set of (int or str)"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_Tuple_1(self):
theirs_tc = pydoc.DocString(func_tuple1).get_type_contract()
ours_tc = pydoc.TypeContract(["tuple"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_NoDocString(self):
theirs_tc = pydoc.DocString(func_no_docstring).get_type_contract()
ours_tc = pydoc.TypeContract(["__NONE__"], ["__NONE__"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_List_2(self):
theirs_tc = pydoc.DocString(func_list2).get_type_contract()
ours_tc = pydoc.TypeContract(["list of (int)"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_WrongFormat_list(self):
theirs_tc = pydoc.DocString(func_wrong_format).get_type_contract()
ours_tc = pydoc.TypeContract(["__ERROR__"], ["__ERROR__"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_Standard_Subclass(self):
theirs_tc = pydoc.DocString(func_standard_subclass).get_type_contract()
ours_tc = pydoc.TypeContract(["io.TextIOWrapper"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_Standard(self):
theirs_tc = pydoc.DocString(func_standard).get_type_contract()
ours_tc = pydoc.TypeContract(["int"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_Mixed1(self):
theirs_tc = pydoc.DocString(func_mixed1).get_type_contract()
ours_tc = pydoc.TypeContract(
["int", "io.TextIOWrapper", "dict", "set", "tuple"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))
def test_TypeContract_Dict_Recursive(self):
theirs_tc = pydoc.DocString(func_dict_recursive).get_type_contract()
ours_tc = pydoc.TypeContract(
["dict of {dict of {dict: dict of {int:str}}:str}"], ["NoneType"])
self.assertEqual(
theirs_tc, ours_tc,
"Type contract does not match. \"{}\" != \"{}\"".format(
theirs_tc, ours_tc))