Skip to content

Commit 8ca93bd

Browse files
committed
bugfix - set homosapiens back as default species in query fucntions
1 parent 4302122 commit 8ca93bd

File tree

4 files changed

+13
-2
lines changed

4 files changed

+13
-2
lines changed

src/tidytcells/mh/_query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717

1818
def query(
19-
species: str,
19+
species: Optional[str] = None,
2020
precision: Optional[Literal["allele", "gene"]] = None,
2121
contains_pattern: Optional[str] = None,
2222
) -> FrozenSet[str]:

src/tidytcells/tr/_query.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818

1919
def query(
20-
species: str,
20+
species: Optional[str] = None,
2121
precision: Optional[str] = None,
2222
functionality: Optional[str] = None,
2323
contains_pattern: Optional[str] = None,

tests/test_mh.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,12 @@ def test_query_contains(
189189
assert expected_in in result
190190
assert not expected_not_in in result
191191

192+
def test_query_default_species(self):
193+
result = mh.query(precision="gene", contains_pattern="DR")
194+
195+
assert len(result) == 10
196+
assert "HLA-DRA" in result
197+
192198

193199
class TestGetChain:
194200
@pytest.mark.parametrize(

tests/test_tr.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -233,6 +233,11 @@ def test_query_functionality(
233233
assert expected_in in result
234234
assert not expected_not_in in result
235235

236+
def test_query_default_species(self):
237+
result = tr.query(precision="gene", contains_pattern="AJ")
238+
assert len(result) == 61
239+
assert "TRAJ11" in result
240+
236241

237242
class TestGetAaSequence:
238243
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)