Fix tests

yethee · May 14, 2024 · 64cf0a3 · 64cf0a3
1 parent def176a
commit 64cf0a3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/Vocab/Loader/DefaultVocabLoaderTest.php b/tests/Vocab/Loader/DefaultVocabLoaderTest.php
@@ -11,6 +11,7 @@
 
 use function copy;
 use function dirname;
+use function file_exists;
 use function file_put_contents;
 use function hash;
 use function rmdir;
@@ -57,8 +58,6 @@ public function testInvalidateCacheWhenChecksumMismatch(): void
     protected function setUp(): void
     {
         $this->cacheDir = sys_get_temp_dir() . '/tiktoken-test';
-
-        self::removeDir($this->cacheDir);
     }
 
     protected function tearDown(): void
@@ -68,6 +67,10 @@ protected function tearDown(): void
 
     private static function removeDir(string $path): void
     {
+        if (! file_exists($path)) {
+            return;
+        }
+
         $iterator = new RecursiveIteratorIterator(
             new RecursiveDirectoryIterator($path, RecursiveDirectoryIterator::SKIP_DOTS),
             RecursiveIteratorIterator::CHILD_FIRST,