Use assert_equal