TST: removed assert in favor of assert_equal