Some tests for next-token prediction