@@ -62,15 +62,18 @@ fn test_lifecycle_basic() {
6262 assert_eq ! ( enc. state( ) , State :: Ready ) ;
6363
6464 // Process should move to Streaming
65- enc. process ( b"test" , & mut [ ] ) . unwrap ( ) ;
65+ enc. process ( b"test" , & mut [ ] )
66+ . expect ( "process should succeed in Ready state" ) ;
6667 assert_eq ! ( enc. state( ) , State :: Streaming ) ;
6768
6869 // Flush should move to Flushing
69- enc. flush ( & mut [ ] ) . unwrap ( ) ;
70+ enc. flush ( & mut [ ] )
71+ . expect ( "flush should succeed in Streaming state" ) ;
7072 assert_eq ! ( enc. state( ) , State :: Flushing ) ;
7173
7274 // Finish should move to Finished
73- enc. finish ( & mut [ ] ) . unwrap ( ) ;
75+ enc. finish ( & mut [ ] )
76+ . expect ( "finish should succeed in Flushing state" ) ;
7477 assert_eq ! ( enc. state( ) , State :: Finished ) ;
7578
7679 // Post-finish calls should error and move to Error
@@ -123,13 +126,19 @@ fn test_huffman_roundtrip_via_traits() {
123126 let input = b"hello world" ;
124127 let mut enc = huffman:: new_encoder ( ) ;
125128 let mut output = vec ! [ 0u8 ; 4096 ] ;
126- enc. process ( input, & mut output) . unwrap ( ) ;
127- let written = enc. finish ( & mut output) . unwrap ( ) ;
129+ enc. process ( input, & mut output)
130+ . expect ( "huffman process should succeed" ) ;
131+ let written = enc
132+ . finish ( & mut output)
133+ . expect ( "huffman finish should succeed" ) ;
128134
129135 let mut dec = huffman:: new_decoder ( ) ;
130136 let mut decoded = vec ! [ 0u8 ; 4096 ] ;
131- dec. process ( & output[ ..written] , & mut decoded) . unwrap ( ) ;
132- let dec_written = dec. finish ( & mut decoded) . unwrap ( ) ;
137+ dec. process ( & output[ ..written] , & mut decoded)
138+ . expect ( "huffman decode process should succeed" ) ;
139+ let dec_written = dec
140+ . finish ( & mut decoded)
141+ . expect ( "huffman decode finish should succeed" ) ;
133142
134143 assert_eq ! ( & decoded[ ..dec_written] , input) ;
135144}
@@ -139,13 +148,19 @@ fn test_arithmetic_roundtrip_via_traits() {
139148 let input = b"test data" ;
140149 let mut enc = arithmetic:: new_encoder ( ) ;
141150 let mut output = vec ! [ 0u8 ; 4096 ] ;
142- enc. process ( input, & mut output) . unwrap ( ) ;
143- let written = enc. finish ( & mut output) . unwrap ( ) ;
151+ enc. process ( input, & mut output)
152+ . expect ( "arithmetic process should succeed" ) ;
153+ let written = enc
154+ . finish ( & mut output)
155+ . expect ( "arithmetic finish should succeed" ) ;
144156
145157 let mut dec = arithmetic:: new_decoder ( ) ;
146158 let mut decoded = vec ! [ 0u8 ; 4096 ] ;
147- dec. process ( & output[ ..written] , & mut decoded) . unwrap ( ) ;
148- let dec_written = dec. finish ( & mut decoded) . unwrap ( ) ;
159+ dec. process ( & output[ ..written] , & mut decoded)
160+ . expect ( "arithmetic decode process should succeed" ) ;
161+ let dec_written = dec
162+ . finish ( & mut decoded)
163+ . expect ( "arithmetic decode finish should succeed" ) ;
149164
150165 assert_eq ! ( & decoded[ ..dec_written] , input) ;
151166}
@@ -155,13 +170,19 @@ fn test_rle_roundtrip_via_traits() {
155170 let input = b"aaabbbccc" ;
156171 let mut enc = rle:: new_encoder ( ) ;
157172 let mut output = vec ! [ 0u8 ; 4096 ] ;
158- enc. process ( input, & mut output) . unwrap ( ) ;
159- let written = enc. finish ( & mut output) . unwrap ( ) ;
173+ enc. process ( input, & mut output)
174+ . expect ( "rle process should succeed" ) ;
175+ let written = enc
176+ . finish ( & mut output)
177+ . expect ( "rle finish should succeed" ) ;
160178
161179 let mut dec = rle:: new_decoder ( ) ;
162180 let mut decoded = vec ! [ 0u8 ; 4096 ] ;
163- dec. process ( & output[ ..written] , & mut decoded) . unwrap ( ) ;
164- let dec_written = dec. finish ( & mut decoded) . unwrap ( ) ;
181+ dec. process ( & output[ ..written] , & mut decoded)
182+ . expect ( "rle decode process should succeed" ) ;
183+ let dec_written = dec
184+ . finish ( & mut decoded)
185+ . expect ( "rle decode finish should succeed" ) ;
165186
166187 assert_eq ! ( & decoded[ ..dec_written] , input) ;
167188}
0 commit comments