@@ -17,6 +17,7 @@ use std::fmt::Display;
17
17
use std:: sync:: Arc ;
18
18
19
19
use databend_common_base:: base:: tokio;
20
+ use databend_common_base:: runtime:: Runtime ;
20
21
use databend_common_exception:: Result ;
21
22
use databend_common_expression:: block_debug:: pretty_format_blocks;
22
23
use databend_common_expression:: DataBlock ;
@@ -62,16 +63,24 @@ async fn test_index_scan_agg_args_are_expression() -> Result<()> {
62
63
test_index_scan_agg_args_are_expression_impl ( "native" ) . await
63
64
}
64
65
65
- #[ tokio:: test( flavor = "multi_thread" ) ]
66
- async fn test_fuzz ( ) -> Result < ( ) > {
67
- test_fuzz_impl ( "parquet" , false ) . await ?;
68
- test_fuzz_impl ( "native" , false ) . await
66
+ #[ test]
67
+ fn test_fuzz ( ) -> Result < ( ) > {
68
+ let runtime = Runtime :: with_worker_threads ( 2 , None ) ?;
69
+ runtime. block_on ( async {
70
+ test_fuzz_impl ( "parquet" , false ) . await ?;
71
+ test_fuzz_impl ( "native" , false ) . await
72
+ } ) ?;
73
+ Ok ( ( ) )
69
74
}
70
75
71
- #[ tokio:: test( flavor = "multi_thread" ) ]
72
- async fn test_fuzz_with_spill ( ) -> Result < ( ) > {
73
- test_fuzz_impl ( "parquet" , true ) . await ?;
74
- test_fuzz_impl ( "native" , true ) . await
76
+ #[ test]
77
+ fn test_fuzz_with_spill ( ) -> Result < ( ) > {
78
+ let runtime = Runtime :: with_worker_threads ( 2 , None ) ?;
79
+ runtime. block_on ( async {
80
+ test_fuzz_impl ( "parquet" , true ) . await ?;
81
+ test_fuzz_impl ( "native" , true ) . await
82
+ } ) ?;
83
+ Ok ( ( ) )
75
84
}
76
85
77
86
async fn plan_sql ( ctx : Arc < QueryContext > , sql : & str ) -> Result < Plan > {
0 commit comments