test/test-dataset.rb in red-datasets-0.1.4 vs test/test-dataset.rb in red-datasets-0.1.5

@@ -1,27 +1,35 @@
 class TestDataset < Test::Unit::TestCase
   sub_test_case("#clear_cache!") do
+    include Helper::PathRestorable
+
     def setup
       @dataset = Datasets::Iris.new
       @cache_dir_path = @dataset.send(:cache_dir_path)
     end
 
     test("when the dataset is downloaded") do
       @dataset.first # This ensures the dataset downloaded
       existence = {before: @cache_dir_path.join("iris.csv").exist?}
 
-      @dataset.clear_cache!
-      existence[:after] = @cache_dir_path.join("iris.csv").exist?
+      restore_path(@cache_dir_path) do
+        @dataset.clear_cache!
+        existence[:after] = @cache_dir_path.join("iris.csv").exist?
 
-      assert_equal({before: true, after: false},
-                   existence)
+        assert_equal({before: true, after: false},
+                     existence)
+      end
     end
 
     test("when the dataset is not downloaded") do
-      FileUtils.rmtree(@cache_dir_path.to_s, secure: true) if @cache_dir_path.exist?
+      restore_path(@cache_dir_path) do
+        if @cache_dir_path.exist?
+          FileUtils.rmtree(@cache_dir_path.to_s, secure: true)
+        end
 
-      assert_nothing_raised do
-        @dataset.clear_cache!
+        assert_nothing_raised do
+          @dataset.clear_cache!
+        end
       end
     end
   end
 end