@@ -11,15 +11,15 @@ import Testing
11
11
12
12
@Suite struct GenFrequencyTests {
13
13
enum Choice : Hashable {
14
- case first
15
- case second ( Int )
16
- case third ( String )
14
+ case plain
15
+ case number ( Int )
16
+ case text ( String )
17
17
}
18
18
19
19
let gen = Gen . oneOf (
20
- Gen . always ( Choice . first ) ,
21
- Gen . int ( ) . map ( Choice . second ) ,
22
- Gen . lowercaseLetter. string ( of: 8 ) . map ( Choice . third ) ,
20
+ Gen . always ( Choice . plain ) ,
21
+ Gen . int ( ) . map ( Choice . number ) ,
22
+ Gen . lowercaseLetter. string ( of: 8 ) . map ( Choice . text ) ,
23
23
)
24
24
25
25
@Test func testGenerateEnum( ) async {
@@ -30,11 +30,11 @@ import Testing
30
30
await confirmation ( expectedCount: 1 ... ) { confirm3 in
31
31
await propertyCheck ( count: 200 , input: gen) { item in
32
32
switch item {
33
- case . first :
33
+ case . plain :
34
34
confirm1 ( )
35
- case . second :
35
+ case . number :
36
36
confirm2 ( )
37
- case . third :
37
+ case . text :
38
38
confirm3 ( )
39
39
}
40
40
}
@@ -44,11 +44,11 @@ import Testing
44
44
}
45
45
46
46
@Test func testShrinkChoice( ) async throws {
47
- let value = ( index: 1 , value: 500 )
48
- let results = Array ( gen. _shrinker ( value) )
47
+ let value = ( index: 1 , value: 500 as Any )
48
+ let results = gen. _shrinker ( value) . compactMap ( gen . _mapFilter )
49
49
try #require( results. count > 1 )
50
- #expect( results. first? . value as? Int == 0 )
51
- #expect( !results. contains { $0 . value as? Int == 500 } )
50
+ #expect( results. first == . number ( 0 ) )
51
+ #expect( !results. contains ( . number ( 500 ) ) )
52
52
}
53
53
}
54
54
#endif // compiler(>=6.2)
0 commit comments