def test_PrintSampleObserver(capsys):
observer = sample_observers.PrintSampleObserver()
sample = model_pb2.Sample(text="Hello, world!")
assert observer.OnSample(sample)
captured = capsys.readouterr()
assert captured.out == """\
def SampleObserversFromFlags() -> typing.List[sample_observers.SampleObserver]:
"""Create sample observers for use with model.Sample() from flags values."""
observers = []
if FLAGS.clgen_min_sample_count >= 0:
app.Warning('--clgen_min_sample_count <= 0 means that sampling (and this '
'process) will never terminate!')
observers.append(
sample_observers.MaxSampleCountObserver(FLAGS.clgen_min_sample_count))
if FLAGS.clgen_cache_sample_protos:
observers.append(sample_observers.LegacySampleCacheObserver())
if FLAGS.clgen_print_samples:
observers.append(sample_observers.PrintSampleObserver())
return observers
def SampleObserversFromFlags():
"""Instantiate sample observers from flag values."""
sample_observers = []
if FLAGS.min_samples <= 0:
app.Warning(
'Entering an infinite sample loop, this process will never end!')
else:
sample_observers.append(
sample_observers_lib.MaxSampleCountObserver(FLAGS.min_samples))
if FLAGS.print_samples:
sample_observers.append(sample_observers_lib.PrintSampleObserver())
if FLAGS.cache_samples:
sample_observers.append(sample_observers_lib.LegacySampleCacheObserver())
if FLAGS.sample_text_dir:
sample_observers.append(
sample_observers_lib.SaveSampleTextObserver(
pathlib.Path(FLAGS.sample_text_dir)))
return sample_observers