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

150
tests/views/test_model_views.py
View File

@ -2,8 +2,8 @@ from django.test import TestCase, override_settings
from django.urls import reverse
from django.conf import settings as s
from epdb.logic import UserManager, PackageManager
from epdb.models import Pathway, Edge, Package, User
from epdb.logic import UserManager
from epdb.models import Package, User
@override_settings(MODEL_DIR=s.FIXTURE_DIRS[0] / "models")
@ -13,10 +13,16 @@ 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.model_package = Package.objects.get(name='Fixtures')
cls.model_package = Package.objects.get(name="Fixtures")
def setUp(self):
self.client.force_login(self.user1)
@ -24,90 +30,96 @@ class PathwayViewTest(TestCase):
def test_predict(self):
self.client.force_login(User.objects.get(username="admin"))
response = self.client.get(
reverse("package model detail", kwargs={
'package_uuid': str(self.model_package.uuid),
'model_uuid': str(self.model_package.models.first().uuid)
}), {
'classify': 'ILikeCats!',
'smiles': 'CCN(CC)C(=O)C1=CC(=CC=C1)CO',
}
reverse(
"package model detail",
kwargs={
"package_uuid": str(self.model_package.uuid),
"model_uuid": str(self.model_package.models.first().uuid),
},
),
{
"classify": "ILikeCats!",
"smiles": "CCN(CC)C(=O)C1=CC(=CC=C1)CO",
},
)
expected = [
{
'products': [
[
'O=C(O)C1=CC(CO)=CC=C1',
'CCNCC'
]
],
'probability': 0.25,
'btrule': {
'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/0e6e9290-b658-4450-b291-3ec19fa19206',
'name': 'bt0430-4011'
}
}, {
'products': [
[
'CCNC(=O)C1=CC(CO)=CC=C1',
'CC=O'
]
], 'probability': 0.0,
'btrule': {
'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/27a3a353-0b66-4228-bd16-e407949e90df',
'name': 'bt0243-4301'
}
}, {
'products': [
[
'CCN(CC)C(=O)C1=CC(C=O)=CC=C1'
]
], 'probability': 0.75,
'btrule': {
'url': 'http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/2f2e0c39-e109-4836-959f-2bda2524f022',
'name': 'bt0001-3568'
}
}
"products": [["O=C(O)C1=CC(CO)=CC=C1", "CCNCC"]],
"probability": 0.25,
"btrule": {
"url": "http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/0e6e9290-b658-4450-b291-3ec19fa19206",
"name": "bt0430-4011",
},
},
{
"products": [["CCNC(=O)C1=CC(CO)=CC=C1", "CC=O"]],
"probability": 0.0,
"btrule": {
"url": "http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/27a3a353-0b66-4228-bd16-e407949e90df",
"name": "bt0243-4301",
},
},
{
"products": [["CCN(CC)C(=O)C1=CC(C=O)=CC=C1"]],
"probability": 0.75,
"btrule": {
"url": "http://localhost:8000/package/1869d3f0-60bb-41fd-b6f8-afa75ffb09d3/simple-ambit-rule/2f2e0c39-e109-4836-959f-2bda2524f022",
"name": "bt0001-3568",
},
},
]
actual = response.json()
self.assertEqual(actual, expected)
response = self.client.get(
reverse("package model detail", kwargs={
'package_uuid': str(self.model_package.uuid),
'model_uuid': str(self.model_package.models.first().uuid)
}), {
'classify': 'ILikeCats!',
'smiles': '',
}
reverse(
"package model detail",
kwargs={
"package_uuid": str(self.model_package.uuid),
"model_uuid": str(self.model_package.models.first().uuid),
},
),
{
"classify": "ILikeCats!",
"smiles": "",
},
)
self.assertEqual(response.status_code, 400)
self.assertEqual(response.json()['error'], 'Received empty SMILES')
self.assertEqual(response.json()["error"], "Received empty SMILES")
response = self.client.get(
reverse("package model detail", kwargs={
'package_uuid': str(self.model_package.uuid),
'model_uuid': str(self.model_package.models.first().uuid)
}), {
'classify': 'ILikeCats!',
'smiles': ' ', # Input should be stripped
}
reverse(
"package model detail",
kwargs={
"package_uuid": str(self.model_package.uuid),
"model_uuid": str(self.model_package.models.first().uuid),
},
),
{
"classify": "ILikeCats!",
"smiles": " ", # Input should be stripped
},
)
self.assertEqual(response.status_code, 400)
self.assertEqual(response.json()['error'], 'Received empty SMILES')
self.assertEqual(response.json()["error"], "Received empty SMILES")
response = self.client.get(
reverse("package model detail", kwargs={
'package_uuid': str(self.model_package.uuid),
'model_uuid': str(self.model_package.models.first().uuid)
}), {
'classify': 'ILikeCats!',
'smiles': 'RandomInput',
}
reverse(
"package model detail",
kwargs={
"package_uuid": str(self.model_package.uuid),
"model_uuid": str(self.model_package.models.first().uuid),
},
),
{
"classify": "ILikeCats!",
"smiles": "RandomInput",
},
)
self.assertEqual(response.status_code, 400)
self.assertEqual(response.json()['error'], '"RandomInput" is not a valid SMILES')
self.assertEqual(response.json()["error"], '"RandomInput" is not a valid SMILES')