Simplify check

MarkusAmshove · Mar 2, 2022 · 11bb0f4 · 11bb0f4
1 parent b898f80
commit 11bb0f4
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/common/src/main/kotlin/org/amshove/kluent/Collections.kt b/common/src/main/kotlin/org/amshove/kluent/Collections.kt
@@ -847,9 +847,7 @@ infix fun <T, I : Iterable<T>> I.shouldContainNone(expected: Array<T>): I =
 infix fun <T, I : Iterable<T>> I.shouldContainNone(check: (T) -> Boolean): I = apply {
     val result = this.map { it to check.invoke(it) }
 
-    if (result.none { it.second }) {
-        Unit
-    } else {
+    if (result.any { it.second }) {
         val failedItems = result
             .filterNot { it.second }
             .map { it.first }
@@ -902,9 +900,7 @@ infix fun <T, I : Iterable<T>> I.shouldNotContainAny(expected: Array<T>): I =
 infix fun <T, I : Iterable<T>> I.shouldNotContainAny(check: (T) -> Boolean): I = apply {
     val result = this.map { it to check.invoke(it) }
 
-    if (result.none { it.second }) {
-        Unit
-    } else {
+    if (result.any { it.second }) {
         val failedItems = result
             .filterNot { it.second }
             .map { it.first }