def test_mixes_streams_with_prob_one(self):
# Mix infinite streams of 0s and 1s.
stream = simple.mix_streams(itertools.repeat(0),
itertools.repeat(1),
mix_prob=1.0)
# Mixed stream should have only 0s.
self.assertEqual(set(itertools.islice(stream, 100)), {0})
def test_mixes_streams_with_prob_zero(self):
stream = simple.mix_streams(
itertools.repeat(0), itertools.repeat(1), mix_prob=0.0)
# Mixed stream should have only 1s.
self.assertEqual(set(itertools.islice(stream, 100)), {1})
def test_mixes_streams_with_prob_half(self):
stream = simple.mix_streams(
itertools.repeat(0), itertools.repeat(1), mix_prob=0.5)
# Mixed stream should have both 0s and 1s.
self.assertEqual(set(itertools.islice(stream, 100)), {0, 1})
def mix_and_batch(streams):
(init_stream, own_stream) = streams
mixed_stream = simple.mix_streams(init_stream, own_stream,
mix_prob)
return simple.batch_stream(mixed_stream,
self._model_train_batch_size)
