Replace assert method in test_set.rb