Add more tests for partial functions