change assertTrue -> assertEqual where fitting