def METHOD_NAME(self):
    self.test_graph = create_graph()
    from neural_compressor.adaptor.tf_utils.graph_rewriter.generic.fetch_weight_from_reshape import (
        FetchWeightFromReshapeOptimizer,
    )
    convert_graph = FetchWeightFromReshapeOptimizer(self.test_graph).do_transformation()
    handled = False
    for node in convert_graph.node:
        if node.op == "Conv2D" and node.input[1] == "reshape/weight_0":
            handled = True
            break
    self.assertEqual(handled, True)