TST: Add small test cases