jebus/enviPy-bayer
1
0
Fork 0
forked from enviPath/enviPy
Code Issues Pull Requests Actions Releases Activity

[Chore] Linted Files (#150)

Browse Source 
Co-authored-by: Tim Lorsbach <tim@lorsba.ch>
Reviewed-on: enviPath/enviPy#150
This commit is contained in:
jebus
2025-10-09 07:25:13 +13:00
parent 22f0bbe10b
commit afeb56622c
50 changed files with 5616 additions and 4408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

78
tests/views/test_pathway_views.py
View File

@ -5,6 +5,7 @@ from django.conf import settings as s
from epdb.logic import UserManager, PackageManager
from epdb.models import Pathway, Edge
@override_settings(MODEL_DIR=s.FIXTURE_DIRS[0] / "models")
class PathwayViewTest(TestCase):
fixtures = ["test_fixtures_incl_model.jsonl.gz"]
@ -12,41 +13,52 @@ class PathwayViewTest(TestCase):
@classmethod
def setUpClass(cls):
super(PathwayViewTest, cls).setUpClass()
cls.user1 = UserManager.create_user("user1", "user1@envipath.com", "SuperSafe",
set_setting=True, add_to_group=True, is_active=True)
cls.user1 = UserManager.create_user(
"user1",
"user1@envipath.com",
"SuperSafe",
set_setting=True,
add_to_group=True,
is_active=True,
)
cls.user1_default_package = cls.user1.default_package
cls.package = PackageManager.create_package(cls.user1, 'Test', 'Test Pack')
cls.package = PackageManager.create_package(cls.user1, "Test", "Test Pack")
def setUp(self):
self.client.force_login(self.user1)
def test_predict_pathway(self):
response = self.client.post(reverse("pathways"), {
'name': 'Test Pathway',
'description': 'Just a Description',
'predict': 'predict',
'smiles': 'CCN(CC)C(=O)C1=CC(=CC=C1)CO',
})
response = self.client.post(
reverse("pathways"),
{
"name": "Test Pathway",
"description": "Just a Description",
"predict": "predict",
"smiles": "CCN(CC)C(=O)C1=CC(=CC=C1)CO",
},
)
self.assertEqual(response.status_code, 302)
pathway_url = response.url
pw = Pathway.objects.get(url=pathway_url)
self.assertEqual(self.user1_default_package, pw.package)
self.assertEqual(pw.name, 'Test Pathway')
self.assertEqual(pw.description, 'Just a Description')
self.assertEqual(pw.name, "Test Pathway")
self.assertEqual(pw.description, "Just a Description")
self.assertEqual(len(pw.root_nodes), 1)
self.assertEqual(pw.root_nodes.first().default_node_label.smiles, 'CCN(CC)C(=O)C1=CC(CO)=CC=C1')
self.assertEqual(
pw.root_nodes.first().default_node_label.smiles, "CCN(CC)C(=O)C1=CC(CO)=CC=C1"
)
first_level_nodes = {
# Edge 1
'CCN(CC)C(=O)C1=CC(C=O)=CC=C1',
"CCN(CC)C(=O)C1=CC(C=O)=CC=C1",
# Edge 2
'CCNC(=O)C1=CC(CO)=CC=C1',
'CC=O',
"CCNC(=O)C1=CC(CO)=CC=C1",
"CC=O",
# Edge 3
'CCNCC',
'O=C(O)C1=CC(CO)=CC=C1',
"CCNCC",
"O=C(O)C1=CC(CO)=CC=C1",
}
predicted_nodes = set()
@ -60,32 +72,36 @@ class PathwayViewTest(TestCase):
def test_predict_package_pathway(self):
response = self.client.post(
reverse("package pathway list", kwargs={'package_uuid': str(self.package.uuid)}), {
'name': 'Test Pathway',
'description': 'Just a Description',
'predict': 'predict',
'smiles': 'CCN(CC)C(=O)C1=CC(=CC=C1)CO',
})
reverse("package pathway list", kwargs={"package_uuid": str(self.package.uuid)}),
{
"name": "Test Pathway",
"description": "Just a Description",
"predict": "predict",
"smiles": "CCN(CC)C(=O)C1=CC(=CC=C1)CO",
},
)
self.assertEqual(response.status_code, 302)
pathway_url = response.url
pw = Pathway.objects.get(url=pathway_url)
self.assertEqual(self.package, pw.package)
self.assertEqual(pw.name, 'Test Pathway')
self.assertEqual(pw.description, 'Just a Description')
self.assertEqual(pw.name, "Test Pathway")
self.assertEqual(pw.description, "Just a Description")
self.assertEqual(len(pw.root_nodes), 1)
self.assertEqual(pw.root_nodes.first().default_node_label.smiles, 'CCN(CC)C(=O)C1=CC(CO)=CC=C1')
self.assertEqual(
pw.root_nodes.first().default_node_label.smiles, "CCN(CC)C(=O)C1=CC(CO)=CC=C1"
)
first_level_nodes = {
# Edge 1
'CCN(CC)C(=O)C1=CC(C=O)=CC=C1',
"CCN(CC)C(=O)C1=CC(C=O)=CC=C1",
# Edge 2
'CCNC(=O)C1=CC(CO)=CC=C1',
'CC=O',
"CCNC(=O)C1=CC(CO)=CC=C1",
"CC=O",
# Edge 3
'CCNCC',
'O=C(O)C1=CC(CO)=CC=C1',
"CCNCC",
"O=C(O)C1=CC(CO)=CC=C1",
}
predicted_nodes = set()