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)