@@ -41,19 +41,19 @@ def test_select_simple(mock_orgdb):
4141def test_select_multikey (mock_orgdb ):
4242 res = mock_orgdb .select (keys = ["1" , "10" ], columns = ["SYMBOL" ], keytype = "ENTREZID" )
4343 assert len (res ) == 2
44-
44+
4545 symbols = res .get_column ("SYMBOL" )
4646 assert "A1BG" in symbols
4747 assert "NAT2" in symbols
4848
4949def test_select_go_expansion (mock_orgdb ):
5050 res = mock_orgdb .select (keys = "1" , columns = ["GO" ], keytype = "ENTREZID" )
51-
51+
5252 col_names = list (res .column_names )
5353 assert "GO" in col_names
5454 assert "EVIDENCE" in col_names
5555 assert "ONTOLOGY" in col_names
56-
56+
5757 assert len (res ) == 2
5858 go_ids = res .get_column ("GO" )
5959 assert "GO:0000001" in go_ids
@@ -66,12 +66,12 @@ def test_select_many_to_one(mock_orgdb):
6666
6767def test_mapIds (mock_orgdb ):
6868 keys = ["1" , "10" , "100" ]
69-
69+
7070 res = mock_orgdb .mapIds (keys , column = "SYMBOL" , keytype = "ENTREZID" )
7171 assert isinstance (res , dict )
7272 assert res ["1" ] == "A1BG"
7373 assert res ["10" ] == "NAT2"
74-
74+
7575 res_list = mock_orgdb .mapIds (["1" ], column = "GO" , keytype = "ENTREZID" , multiVals = "list" )
7676 assert isinstance (res_list ["1" ], list )
7777 assert len (res_list ["1" ]) == 2
@@ -81,12 +81,12 @@ def test_genes_genomicranges(mock_orgdb):
8181 gr = mock_orgdb .genes ()
8282 assert isinstance (gr , GenomicRanges )
8383 assert len (gr ) == 2
84-
84+
8585 names = list (gr .names )
8686 idx = names .index ("1" )
8787 assert str (gr .seqnames [idx ]) == "chr19"
8888 assert gr .start [idx ] == 58346806
8989 assert gr .end [idx ] == 58353492
90-
90+
9191 assert "gene_id" in gr .mcols .column_names
92- assert gr .mcols .get_column ("gene_id" )[idx ] == "1"
92+ assert gr .mcols .get_column ("gene_id" )[idx ] == "1"
0 commit comments