Force batch_size=4 for this example.
Seed random number generator to avoid nondeterministic tests.
Rename dataset variables and flesh out docstring.