test.py: optionally just use the 10k validation set

This makes the development cycle a bit quicker.