Skip to content

Commit

Permalink
Updated test_anndata_enricher.py test_graph_generator.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ubyndr committed Feb 5, 2024
1 parent e7413de commit 86ac860
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 8 deletions.
18 changes: 11 additions & 7 deletions test/graph_generator/test_graph_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,9 @@ def graph_generator_instance_for_kidney(enrichment_analyzer_instance_for_kidney_
return GraphGenerator(ea)


def test_graph_generator_init_missing_enrichment_process(enrichment_analyzer_instance_for_immune_data):
def test_graph_generator_init_missing_enrichment_process(
enrichment_analyzer_instance_for_immune_data,
):
ea = enrichment_analyzer_instance_for_immune_data
ea.co_annotation_report()

Expand All @@ -72,7 +74,9 @@ def test_graph_generator_init_missing_enrichment_process(enrichment_analyzer_ins
assert exception.args[0] == expected_message


def test_graph_generator_init_missing_analysis_process(enrichment_analyzer_instance_for_immune_data):
def test_graph_generator_init_missing_analysis_process(
enrichment_analyzer_instance_for_immune_data,
):
ea = enrichment_analyzer_instance_for_immune_data
ea.enricher_manager.simple_enrichment()

Expand Down Expand Up @@ -119,7 +123,7 @@ def test_graph_generator_init_with_valid_input(enrichment_analyzer_instance_for_
def test_generate_rdf_graph(graph_generator_instance_for_kidney):
graph_generator = graph_generator_instance_for_kidney
graph_generator.generate_rdf_graph()
assert len(graph_generator.graph) == 441
assert len(graph_generator.graph) == 433
assert (
len([[s, p, o] for s, p, o in graph_generator.graph.triples((None, RDF.type, None))]) == 95
)
Expand All @@ -135,7 +139,7 @@ def test_generate_rdf_graph(graph_generator_instance_for_kidney):
)
]
)
== 85
== 77
)
assert (
len(
Expand All @@ -161,11 +165,11 @@ def test_enrich_rdf_graph(graph_generator_instance_for_kidney):
graph_generator = graph_generator_instance_for_kidney
graph_generator.generate_rdf_graph()

assert len(graph_generator.graph) == 441
assert len(graph_generator.graph) == 433

graph_generator.enrich_rdf_graph()

assert len(graph_generator.graph) == 1024
assert len(graph_generator.graph) == 903
assert (
URIRef(CONSIST_OF.get("iri")),
RDFS.label,
Expand All @@ -182,7 +186,7 @@ def test_enrich_rdf_graph(graph_generator_instance_for_kidney):
if str(s).startswith("http://purl.obolibrary.org/obo/CL_")
]
)
== 530
== 417
)


Expand Down
10 changes: 9 additions & 1 deletion test/test_anndata_enricher.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,15 @@ def slim_data():
]


def test_init_defaults(sample_immune_data, slim_data):
def test_init_defaults(mocker, sample_immune_data, slim_data):
iterable_slim_data = slim_data
mocker.patch.object(
SlimManager,
"get_slim_list",
side_effect=[
iter(iterable_slim_data),
],
)
# Test initialization with default values
enricher = AnndataEnricher(sample_immune_data)

Expand Down

0 comments on commit 86ac860

Please sign in to comment.