replace assert with assert_equal