1 files changed, 9 insertions, 8 deletions
diff --git a/SoraTests/FolderHierarchyTests.swift b/SoraTests/FolderHierarchyTests.swift
index 6bf7f4a..911b8f4 100644
--- a/SoraTests/FolderHierarchyTests.swift
+++ b/SoraTests/FolderHierarchyTests.swift
@@ -1,5 +1,6 @@
-import XCTest
+import Nimble
 @testable import Sora
+import XCTest
 
 final class FolderHierarchyTests: XCTestCase {
   func testTopLevelFolderIncludesRootFolderAndNestedChildren() {
@@ -17,10 +18,10 @@ final class FolderHierarchyTests: XCTestCase {
     )
     let hierarchy = FolderHierarchy(folders: [rootFolder, nestedFolder, unrelatedFolder])
 
-    XCTAssertTrue(hierarchy.isInTopLevelFolder(rootFolder.id, named: "Artists"))
-    XCTAssertTrue(hierarchy.isInTopLevelFolder(nestedFolder.id, named: "Artists"))
-    XCTAssertFalse(hierarchy.isInTopLevelFolder(unrelatedFolder.id, named: "Artists"))
-    XCTAssertFalse(hierarchy.isInTopLevelFolder(nil, named: "Artists"))
+    expect(hierarchy.isInTopLevelFolder(rootFolder.id, named: "Artists")) == true
+    expect(hierarchy.isInTopLevelFolder(nestedFolder.id, named: "Artists")) == true
+    expect(hierarchy.isInTopLevelFolder(unrelatedFolder.id, named: "Artists")) == false
+    expect(hierarchy.isInTopLevelFolder(nil, named: "Artists")) == false
   }
 
   func testTopLevelUncategorizedOnlyMatchesRootFolder() {
@@ -34,8 +35,8 @@ final class FolderHierarchyTests: XCTestCase {
     )
     let hierarchy = FolderHierarchy(folders: [rootFolder, nestedFolder])
 
-    XCTAssertTrue(hierarchy.isInTopLevelUncategorized(rootFolder.id, named: "Artists"))
-    XCTAssertFalse(hierarchy.isInTopLevelUncategorized(nestedFolder.id, named: "Artists"))
-    XCTAssertFalse(hierarchy.isInTopLevelUncategorized(nil, named: "Artists"))
+    expect(hierarchy.isInTopLevelUncategorized(rootFolder.id, named: "Artists")) == true
+    expect(hierarchy.isInTopLevelUncategorized(nestedFolder.id, named: "Artists")) == false
+    expect(hierarchy.isInTopLevelUncategorized(nil, named: "Artists")) == false
   }
 }