Improve some tests using assertIs (#329)

2 files changed