def METHOD_NAME(self):
    from package_a.test_module import SimpleTest
    model = SimpleTest()
    f = BytesIO()
    with PackageExporter(f) as pe:
        pe.intern("**")
        pe.save_pickle("model", "model.pkl", model)
    f.seek(0)
    pi = PackageImporter(f)
    loaded = pi.load_pickle("model", "model.pkl")
    traced = symbolic_trace(loaded)
    input = torch.rand(2, 3)
    self.assertEqual(loaded(input), traced(input))