def METHOD_NAME(cls):
    cls.pbtxts = [
        os.path.join(tests_path, "infer/fparam_aparam.pbtxt"),
    ]
    cls.graph_dirs = [pbtxt.replace("pbtxt", "pb") for pbtxt in cls.pbtxts]
    for pbtxt, pb in zip(cls.pbtxts, cls.graph_dirs):
        convert_pbtxt_to_pb(pbtxt, pb)
    cls.graphs = [DeepPotential(pb) for pb in cls.graph_dirs]