Skip to content

Commit 7f20056

Browse files
Update model names in consensus test and remove direct test
Replaces model names in test_consensus.py with updated versions and deletes the test_direct.py file, which contained utility function tests. This streamlines the test suite and ensures model references are current.
1 parent e4ea7d1 commit 7f20056

2 files changed

Lines changed: 7 additions & 159 deletions

File tree

python/tests/test_consensus.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -26,17 +26,17 @@ def setup(self, sample_marker_genes_df, sample_marker_genes_dict):
2626

2727
# Sample annotations from different models
2828
self.model_annotations = {
29-
"gpt-4o": { # Using real model name
29+
"gpt-5": {
3030
"1": "T cells",
3131
"2": "B cells",
3232
"3": "NK cells",
3333
},
34-
"claude-3-opus": { # Using real model name
34+
"claude-sonnet-4-5-20250929": {
3535
"1": "T lymphocytes",
3636
"2": "B lymphocytes",
3737
"3": "Natural killer cells",
3838
},
39-
"gemini-1.5-pro": { # Using real model name
39+
"gemini-3-pro": {
4040
"1": "CD4+ T cells",
4141
"2": "Plasma B cells",
4242
"3": "NK cells",
@@ -136,10 +136,10 @@ def test_interactive_consensus_annotation(
136136
marker_genes=self.marker_genes_dict,
137137
species="human",
138138
models=[
139-
"gpt-4o",
140-
"claude-3-opus",
141-
"gemini-1.5-pro",
142-
], # Using real model names
139+
"gpt-5",
140+
"claude-sonnet-4-5-20250929",
141+
"gemini-3-pro",
142+
],
143143
api_keys={
144144
"openai": "test-key",
145145
"anthropic": "test-key",

python/tests/test_direct.py

Lines changed: 0 additions & 152 deletions
This file was deleted.

0 commit comments

Comments
 (0)