def METHOD_NAME(mode):
    ms.set_context(mode=mode)
    np.random.seed(0)
    net = SimpleCNNWithInnerSequential()
    x = ms.Tensor(np.random.randn(8, 1, 32, 32), dtype=ms.float32)
    y0 = net.forward_multi_features(x)
    wrapped_net = FeatureExtractWrapper(net, [0, 1, 2])
    y1 = wrapped_net(x)
    assert len(y0) == len(y1)
    assert len(y1) == 3
    for z0, z1 in zip(y0, y1):
        np.testing.assert_equal(z1.asnumpy(), z0.asnumpy())