use assert_equal in test