def test_approximate_unique_combine_fn_requires_compatible_coder(self):
test_input = 'a'
sample_size = 30
coder = coders.FloatCoder()
combine_fn = ApproximateUniqueCombineFn(sample_size, coder)
accumulator = combine_fn.create_accumulator()
with self.assertRaises(RuntimeError) as e:
accumulator = combine_fn.add_input(accumulator, test_input)
self.assertRegex(e.exception.args[0], 'Runtime exception')
def test_approximate_unique_combine_fn_requires_nondeterministic_coder(self):
sample_size = 30
coder = coders.Base64PickleCoder()
with self.assertRaises(ValueError) as e:
_ = ApproximateUniqueCombineFn(sample_size, coder)
self.assertRegex(
e.exception.args[0],
'The key coder "Base64PickleCoder" '
'for ApproximateUniqueCombineFn is not deterministic.')