spec/sequel_spec.rb in tqdm-0.3.0 vs spec/sequel_spec.rb in tqdm-0.4.0

@@ -36,9 +36,19 @@
           "\r|######----| 3/5  60% [elapsed: 00:03 left: 00:02,  1.00 iters/sec]" \
           "\r|########--| 4/5  80% [elapsed: 00:04 left: 00:01,  1.00 iters/sec]" \
           "\r                                                                   " \
           "\r"
       end
+
+      it 'returns a re-frozen object' do
+        enhanced = timecop_loop(database[:items])
+        expect(enhanced.frozen?).to be_truthy
+      end
+
+      it 'returns an object inheriting from Sequel::Dataset' do
+        enhanced = timecop_loop(database[:items])
+        expect(enhanced).to be_kind_of(Sequel::Dataset)
+      end
     end
     
     context 'with leave: true' do
       it 'displays a progress bar with as many steps as elements and leaves it' do
         final_stderr = with_stderr { timecop_loop(database[:items], leave: true) }