Merge pull request #763 from fartem/tests-4

2024-11-01 v. 6.9.4.1: updated tests for "380. Insert Delete GetRandom O(1)"

Author: fartem

leetcode-ruby.gemspec

@@ -5,7 +5,7 @@ require 'English'
 ::Gem::Specification.new do |s|
   s.required_ruby_version = '>= 3.0'
   s.name = 'leetcode-ruby'
-  s.version = '6.9.4'
+  s.version = '6.9.4.1'
   s.license = 'MIT'
   s.files = ::Dir['lib/**/*.rb'] + %w[README.md]
   s.executable = 'leetcode-ruby'

test/medium/test_380_insert_delete_getrandom_o1.rb

@@ -10,7 +10,10 @@ def test_default_one
     assert(randomized_set.insert(1))
     assert(!randomized_set.remove(2))
     assert(randomized_set.insert(2))
-    assert(randomized_set.get_random == 1 || randomized_set.get_random == 2)
+
+    rand = randomized_set.get_random
+    assert([1, 2].include?(rand))
+
     assert(randomized_set.remove(1))
     assert(!randomized_set.insert(2))
     assert_equal(2, randomized_set.get_random)