如何在Apache横梁单元测试中正确测试pcollection长度

import org.apache.beam.sdk.testing.PAssert;
import org.apache.beam.sdk.testing.TestPipeline;
import org.apache.beam.sdk.values.PCollection;
import org.junit.Rule;
import org.junit.Test;

public class MyTest {
    @Rule
    public final transient TestPipeline pipeline = TestPipeline.create();
    
    // 测试方法
}

@Test
public void testPCollectionLength() {
    // 构建测试数据
    PCollection<Integer> input = pipeline.apply(Create.of(1, 2, 3, 4, 5));
    
    // 执行你的转换操作
    PCollection<Integer> output = input.apply(YourTransform.create());
    
    // 使用PAssert验证PCollection的长度
    PAssert.that(output).satisfies(collection -> {
        // 获取PCollection的元素列表
        List<Integer> elements = Lists.newArrayList(collection);
        
        // 验证PCollection的长度是否符合预期
        assertThat(elements.size(), equalTo(5)); // 例如，预期长度为5
        
        return null;
    });
    
    // 运行测试管道
    pipeline.run().waitUntilFinish();
}