From 0c2228e699d515bd111ca8154f35e0711108d2fc Mon Sep 17 00:00:00 2001 From: fartem Date: Fri, 1 Nov 2024 09:08:15 +0300 Subject: [PATCH] 2024-11-01 v. 6.9.4.1: updated tests for "380. Insert Delete GetRandom O(1)" --- leetcode-ruby.gemspec | 2 +- test/medium/test_380_insert_delete_getrandom_o1.rb | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec index bf51dc15..c2cc5f5b 100644 --- a/leetcode-ruby.gemspec +++ b/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' diff --git a/test/medium/test_380_insert_delete_getrandom_o1.rb b/test/medium/test_380_insert_delete_getrandom_o1.rb index b646e7a7..7a8e35c3 100644 --- a/test/medium/test_380_insert_delete_getrandom_o1.rb +++ b/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)