def test_size(self):
cache = StreamingCache(cache_dir=None)
cache.write([TestStreamFileRecord()], 'my_label')
coder = cache.load_pcoder('my_label')
# Add one because of the new-line character that is also written.
size = len(coder.encode(TestStreamFileRecord().SerializeToString())) + 1
self.assertEqual(cache.size('my_label'), size)
def test_always_default_coder_for_test_stream_records(self):
CACHED_NUMBERS = repr(CacheKey('numbers', '', '', ''))
numbers = (FileRecordsBuilder(CACHED_NUMBERS)
.advance_processing_time(2)
.add_element(element=1, event_time_secs=0)
.advance_processing_time(1)
.add_element(element=2, event_time_secs=0)
.advance_processing_time(1)
.add_element(element=2, event_time_secs=0)
.build()) # yapf: disable
cache = StreamingCache(cache_dir=None)
cache.write(numbers, CACHED_NUMBERS)
self.assertIs(type(cache.load_pcoder(CACHED_NUMBERS)),
type(cache._default_pcoder))
