Change tests to use assertEqual

Signed-off-by: Nicolas Iooss <nicolas.iooss_git@polytechnique.org>