Coverage for tests/test_statistics.py: 94%

1import unittest 

2 

3from rulekit.main import RuleKit 

4from rulekit import classification 

5from rulekit.rules import RuleSetStatistics 

6from sklearn.datasets import load_iris 

7 

8 

9class TestRuleClassifier(unittest.TestCase): 

10 

11 @classmethod 

12 def setUpClass(cls): 

13 RuleKit.init() 

14 

15 def test_classification_accuracy_on_iris(self): 

16 clf = classification.RuleClassifier() 

17 x, y = load_iris(return_X_y=True) 

18 

19 clf.fit(x, y) 

20 

21 ruleset_stats: RuleSetStatistics = clf.model.stats 

22 for rule in clf.model.rules: 

23 rule.stats 

24 

25 

26if __name__ == '__main__': 

27 unittest.main()