Skip to content

Commit

Permalink
Add missing eggnog test
Browse files Browse the repository at this point in the history
  • Loading branch information
misialq committed May 3, 2024
1 parent 1069039 commit 53b2690
Showing 1 changed file with 23 additions and 1 deletion.
24 changes: 23 additions & 1 deletion q2_moshpit/eggnog/tests/test_method.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,7 @@ def test_eggnog_search_parallel_contigs(self):

pdt.assert_frame_equal(parallel, single)

def test_eggnog_search_parallel_mags(self):
def test_eggnog_search_parallel_mags_derep(self):
mags = qiime2.Artifact.import_data(
'FeatureData[MAG]',
self.get_data_path('mag-sequences')
Expand All @@ -146,6 +146,28 @@ def test_eggnog_search_parallel_mags(self):

pdt.assert_frame_equal(parallel, single)

def test_eggnog_search_parallel_mags(self):
mags = qiime2.Artifact.import_data(
'SampleData[MAGs]',
self.get_data_path('mag-sequences-per-sample')
)

with ParallelConfig():
_, parallel = self.eggnog_diamond_search.parallel(
mags,
self.diamond_db_artifact
)._result()

_, single = self._eggnog_diamond_search(
sequences=mags,
diamond_db=self.diamond_db_artifact
)

parallel = parallel.view(pd.DataFrame)
single = single.view(pd.DataFrame)

pdt.assert_frame_equal(parallel, single)


class TestAnnotate(TestPluginBase):
package = 'q2_moshpit.eggnog.tests'
Expand Down

0 comments on commit 53b2690

Please sign in to comment.