TST: add a test