chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,50 @@
|
||||
# Copyright (c) Microsoft. All rights reserved.
|
||||
|
||||
from datetime import timedelta
|
||||
|
||||
from semantic_kernel.agents.open_ai.run_polling_options import RunPollingOptions
|
||||
|
||||
|
||||
def test_get_polling_interval_below_threshold():
|
||||
options = RunPollingOptions()
|
||||
iteration_count = 1
|
||||
expected_interval = timedelta(milliseconds=250)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
|
||||
def test_get_polling_interval_at_threshold():
|
||||
options = RunPollingOptions()
|
||||
iteration_count = 2
|
||||
expected_interval = timedelta(milliseconds=250)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
|
||||
def test_get_polling_interval_above_threshold():
|
||||
options = RunPollingOptions()
|
||||
iteration_count = 3
|
||||
expected_interval = timedelta(seconds=1)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
|
||||
def test_get_polling_interval_custom_threshold():
|
||||
options = RunPollingOptions(run_polling_backoff_threshold=5)
|
||||
iteration_count = 4
|
||||
expected_interval = timedelta(milliseconds=250)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
iteration_count = 6
|
||||
expected_interval = timedelta(seconds=1)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
|
||||
def test_get_polling_interval_custom_intervals():
|
||||
options = RunPollingOptions(
|
||||
run_polling_interval=timedelta(milliseconds=500), run_polling_backoff=timedelta(seconds=2)
|
||||
)
|
||||
iteration_count = 1
|
||||
expected_interval = timedelta(milliseconds=500)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
|
||||
iteration_count = 3
|
||||
expected_interval = timedelta(seconds=2)
|
||||
assert options.get_polling_interval(iteration_count) == expected_interval
|
||||
Reference in New Issue
Block a user