Total Complexity | 3 |
Total Lines | 34 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | import pytest |
||
2 | from chembl_webresource_client.new_client import new_client as Chembl |
||
3 | |||
4 | from mandos.model.targets import TargetFactory |
||
5 | from mandos.search.chembl.target_traversal_strategy import ( |
||
6 | TargetTraversalStrategies, |
||
7 | ) |
||
8 | |||
9 | |||
10 | class TestTargetTraversalStrategy1: |
||
11 | # def test_gabaa(self): |
||
12 | # target = TargetFactory.find("CHEMBL2109243", Chembl) |
||
13 | # found = TargetTraversalStrategies.strategy1(Chembl).traverse(target) |
||
14 | # assert [f.chembl for f in found] == ["CHEMBL2093872"] |
||
15 | |||
16 | def test_5ht2b(self): |
||
17 | target = TargetFactory.find("CHEMBL1833", Chembl) |
||
18 | found = TargetTraversalStrategies.strategy1(Chembl).traverse(target) |
||
19 | assert [f.chembl for f in found] == ["CHEMBL2096904"] |
||
20 | |||
21 | def test_5ht2bc_sel_group(self): |
||
22 | target = TargetFactory.find("CHEMBL2111466", Chembl) |
||
23 | found = TargetTraversalStrategies.strategy1(Chembl).traverse(target) |
||
24 | assert [f.chembl for f in found] == ["CHEMBL2096904"] |
||
25 | |||
26 | def test_mu_or(self): |
||
27 | target = TargetFactory.find("CHEMBL233", Chembl) |
||
28 | found = TargetTraversalStrategies.strategy1(Chembl).traverse(target) |
||
29 | assert [f.chembl for f in found] == ["CHEMBL2095181"] |
||
30 | |||
31 | |||
32 | if __name__ == "__main__": |
||
33 | pytest.main() |
||
34 |