From 7569c20454a1beb3a494f6a024839f49d73f6447 Mon Sep 17 00:00:00 2001 From: yangxuan Date: Mon, 11 Nov 2024 18:46:53 +0800 Subject: [PATCH] fix Signed-off-by: yangxuan --- tests/test_rate_runner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_rate_runner.py b/tests/test_rate_runner.py index 943d7886..363b7a33 100644 --- a/tests/test_rate_runner.py +++ b/tests/test_rate_runner.py @@ -30,7 +30,7 @@ def test_rate_runner(db, insert_rate): _, t = runner.run_with_rate() log.info(f"insert run done, time={t}") -def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[float], stage_search_dur: int, local: bool=False): +def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[float], read_dur_after_write: int, local: bool=False): cohere = Dataset.COHERE.manager(1_000_000) if local is True: source = DatasetSource.AliyunOSS @@ -44,7 +44,7 @@ def test_read_write_runner(db, insert_rate, conc: list, search_stage: Iterable[f dataset=cohere, insert_rate=insert_rate, search_stage=search_stage, - stage_search_dur=stage_search_dur, + read_dur_after_write=read_dur_after_write, concurrencies=conc ) rw_runner.run_read_write()