[tests] IRR-NCDM

bigdata-ustc · Jul 1, 2021 · 4bcebd3 · 4bcebd3
1 parent 2bc3a34
commit 4bcebd3
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/irr/test_ncdm.py b/tests/irr/test_ncdm.py
@@ -0,0 +1,20 @@
+# coding: utf-8
+# 2021/7/1 @ tongshiwei
+
+from EduCDM.IRR import NCDM
+
+
+def test_irr_dina(train_data, test_data, params, tmp_path):
+    cdm = NCDM(params.user_num, params.item_num, params.knowledge_num)
+    cdm.train(train_data, test_data=test_data, epoch=2)
+    filepath = tmp_path / "irr.params"
+    cdm.save(filepath)
+    cdm.load(filepath)
+
+
+def test_irt(zero_train_data, test_data, params, tmp_path):
+    cdm = NCDM(params.user_num, params.item_num, params.knowledge_num, zeta=0)
+    cdm.train(zero_train_data, test_data=test_data, epoch=2)
+    filepath = tmp_path / "irr.params"
+    cdm.save(filepath)
+    cdm.load(filepath)