def assertBuiltinsPickleEqual(self, f1, f2): with open(f1, "rb") as pickle1, open(f2, "rb") as pickle2: if pickle1.read() == pickle2.read(): return out1 = pytd_utils.LoadPickle(f1, compress=True) out2 = pytd_utils.LoadPickle(f2, compress=True) raise AssertionError("\n".join(pytd_utils.DiffNamedPickles(out1, out2)))
def testDiffSamePickle(self): ast = pytd.TypeDeclUnit("foo", (), (), (), (), ()) with file_utils.Tempdir() as d: filename = os.path.join(d.path, "foo.pickled") serialize_ast.StoreAst(ast, filename) with open(filename, "rb") as fi: data = fi.read() named_pickles = [("foo", data)] self.assertFalse(pytd_utils.DiffNamedPickles(named_pickles, named_pickles))
def test_diff_pickle_length(self): ast = pytd.TypeDeclUnit("foo", (), (), (), (), ()) with file_utils.Tempdir() as d: filename = os.path.join(d.path, "foo.pickled") serialize_ast.StoreAst(ast, filename) with open(filename, "rb") as fi: data = fi.read() named_pickles1 = [] named_pickles2 = [("foo", data)] self.assertTrue(pytd_utils.DiffNamedPickles(named_pickles1, named_pickles2))
def testDiffPickleAst(self): ast1 = pytd.TypeDeclUnit("foo", (), (), (), (), ()) ast2 = ast1.Replace(type_params=(pytd.TypeParameter("T", (), None, None),)) with file_utils.Tempdir() as d: data = [] for ast in (ast1, ast2): filename = os.path.join(d.path, "foo.pickled") serialize_ast.StoreAst(ast, filename) with open(filename, "rb") as fi: data.append(fi.read()) named_pickles1 = [("foo", data[0])] named_pickles2 = [("foo", data[1])] self.assertTrue(pytd_utils.DiffNamedPickles(named_pickles1, named_pickles2))