Skip to content

Commit 0fdc4c5

Browse files
committed
Run black code formatter
1 parent b0fda23 commit 0fdc4c5

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

sdk/evaluation/azure-ai-evaluation/tests/unittests/test_redteam_user_agent.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -10,78 +10,78 @@ class TestRedTeamUserAgent:
1010
def test_default_redteam_context(self):
1111
"""Test that the default red team context works as expected."""
1212
base_user_agent = UserAgentSingleton().value
13-
13+
1414
with UserAgentSingleton.redteam_context():
1515
redteam_user_agent = UserAgentSingleton().value
1616
expected = f"{base_user_agent} (type=redteam; subtype=RedTeam)"
1717
assert redteam_user_agent == expected
18-
18+
1919
# Should restore to original after context
2020
assert UserAgentSingleton().value == base_user_agent
2121

2222
def test_custom_subtype_redteam_context(self):
2323
"""Test red team context with custom subtype."""
2424
base_user_agent = UserAgentSingleton().value
2525
custom_subtype = "CustomTest"
26-
26+
2727
with UserAgentSingleton.redteam_context(subtype=custom_subtype):
2828
redteam_user_agent = UserAgentSingleton().value
2929
expected = f"{base_user_agent} (type=redteam; subtype={custom_subtype})"
3030
assert redteam_user_agent == expected
31-
31+
3232
# Should restore to original after context
3333
assert UserAgentSingleton().value == base_user_agent
3434

3535
def test_nested_contexts(self):
3636
"""Test nested user agent contexts."""
3737
base_user_agent = UserAgentSingleton().value
38-
38+
3939
with UserAgentSingleton.add_useragent_product("outer/1.0"):
4040
outer_user_agent = UserAgentSingleton().value
4141
expected_outer = f"{base_user_agent} outer/1.0"
4242
assert outer_user_agent == expected_outer
43-
43+
4444
with UserAgentSingleton.redteam_context(subtype="NestedTest"):
4545
nested_user_agent = UserAgentSingleton().value
4646
expected_nested = f"{expected_outer} (type=redteam; subtype=NestedTest)"
4747
assert nested_user_agent == expected_nested
48-
48+
4949
# Should restore to outer context
5050
assert UserAgentSingleton().value == expected_outer
51-
51+
5252
# Should restore to original after all contexts
5353
assert UserAgentSingleton().value == base_user_agent
5454

5555
def test_multiple_redteam_contexts(self):
5656
"""Test multiple red team contexts in sequence."""
5757
base_user_agent = UserAgentSingleton().value
58-
58+
5959
# First context
6060
with UserAgentSingleton.redteam_context(subtype="First"):
6161
first_user_agent = UserAgentSingleton().value
6262
expected_first = f"{base_user_agent} (type=redteam; subtype=First)"
6363
assert first_user_agent == expected_first
64-
64+
6565
# Should restore to original
6666
assert UserAgentSingleton().value == base_user_agent
67-
68-
# Second context
67+
68+
# Second context
6969
with UserAgentSingleton.redteam_context(subtype="Second"):
7070
second_user_agent = UserAgentSingleton().value
7171
expected_second = f"{base_user_agent} (type=redteam; subtype=Second)"
7272
assert second_user_agent == expected_second
73-
73+
7474
# Should restore to original
7575
assert UserAgentSingleton().value == base_user_agent
7676

7777
def test_redteam_context_kwargs_ignored(self):
7878
"""Test that additional kwargs are ignored gracefully."""
7979
base_user_agent = UserAgentSingleton().value
80-
80+
8181
with UserAgentSingleton.redteam_context(subtype="Test", ignored_param="should_be_ignored"):
8282
redteam_user_agent = UserAgentSingleton().value
8383
expected = f"{base_user_agent} (type=redteam; subtype=Test)"
8484
assert redteam_user_agent == expected
85-
85+
8686
# Should restore to original after context
87-
assert UserAgentSingleton().value == base_user_agent
87+
assert UserAgentSingleton().value == base_user_agent

0 commit comments

Comments
 (0)