rename test function

bigdata-ustc · Oct 22, 2021 · 83b7776 · 83b7776
1 parent 416dfdb
commit 83b7776
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/EduCDM/IRT/GD/IRT.py b/EduCDM/IRT/GD/IRT.py
@@ -9,7 +9,7 @@
 import torch.nn.functional as F
 from tqdm import tqdm
 from ..irt import irt3pl
-from sklearn.metrics import roc_auc_score, accuracy_score, classification_report
+from sklearn.metrics import roc_auc_score, accuracy_score
 
 
 class IRTNet(nn.Module):

diff --git a/tests/irt/gd/test_gdirt.py b/tests/irt/gd/test_gdirt.py
@@ -14,7 +14,7 @@ def test_train(data, conf, tmp_path):
     cdm.load(filepath)
 
 
-def test_raises(data, conf, tmp_path):
+def test_exception(data, conf, tmp_path):
     try:
         user_num, item_num = conf
         cdm = GDIRT(user_num, item_num, value_range=10, a_range=100)

diff --git a/tests/mirt/test_mirt.py b/tests/mirt/test_mirt.py
@@ -14,7 +14,7 @@ def test_train(data, conf, tmp_path):
     cdm.load(filepath)
 
 
-def test_raises(data, conf, tmp_path):
+def test_exception(data, conf, tmp_path):
     try:
         user_num, item_num = conf
         cdm = MIRT(user_num, item_num, 10, a_range=100)