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

start towards #120

Browse Source
This commit is contained in:
Liam Brydon
2025-10-22 08:22:29 +13:00
parent 376fd65785
commit 2980a75daa
3 changed files with 53 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_dataset.py
View File

@ -2,7 +2,7 @@ from django.test import TestCase
from epdb.logic import PackageManager
from epdb.models import Reaction, Compound, User, Rule
from utilities.ml import Dataset
from utilities.ml import RuleBasedDataset
class DatasetTest(TestCase):
@ -46,7 +46,7 @@ class DatasetTest(TestCase):
reactions = [r for r in Reaction.objects.filter(package=self.package)]
applicable_rules = [self.rule1]
ds = Dataset.generate_dataset(reactions, applicable_rules)
ds = RuleBasedDataset.generate_dataset(reactions, applicable_rules)
self.assertEqual(len(ds.y()), 1)
self.assertEqual(sum(ds.y()[0]), 1)