TST: add more tests but not all pass