CLN: clean up test_ujson.py a bit