Skip to content

Commit fccc9c5

Browse files
authored
Fix Optional generator returning more nils when value is filtered
1 parent 4b7809b commit fccc9c5

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

Sources/PropertyBased/Generator.swift

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -233,11 +233,13 @@ extension Generator {
233233
Shrink.WithNil(nil)
234234
}
235235
},
236-
finalResult: {
237-
if let some = $0 {
238-
return self._mapFilter(some)
236+
finalResult: { value in
237+
guard let value else {
238+
return .some(.none)
239239
}
240-
return .some(.none)
240+
241+
let filtered = self._mapFilter(value)
242+
return filtered.flatMap { .some($0) }
241243
})
242244
}
243245

0 commit comments

Comments
 (0)