apache_beam.runners.interactive.testing.test_cache_manager module

class apache_beam.runners.interactive.testing.test_cache_manager.InMemoryCache[source]

Bases: apache_beam.runners.interactive.cache_manager.CacheManager

A cache that stores all PCollections in an in-memory map.

This is only used for checking the pipeline shape. This can’t be used for running the pipeline isn’t shared between the SDK and the Runner.

exists(*labels)[source]
read(*labels, **args)[source]
write(value, *labels)[source]
save_pcoder(pcoder, *labels)[source]
load_pcoder(*labels)[source]
cleanup()[source]
clear(*label)[source]
source(*labels)[source]
sink(labels, is_capture=False)[source]
size(*labels)[source]
class apache_beam.runners.interactive.testing.test_cache_manager.NoopSink(label=None)[source]

Bases: apache_beam.transforms.ptransform.PTransform

expand(pcoll)[source]
class apache_beam.runners.interactive.testing.test_cache_manager.FileRecordsBuilder(tag=None)[source]

Bases: object

add_element(element, event_time_secs)[source]
advance_watermark(watermark_secs)[source]
advance_processing_time(delta_secs)[source]
build()[source]