test: add unit tests for gowalla, tafeng, and yoochoose datasets

Author: hieuddo

tests/cornac/datasets/test_gowalla.py

@@ -0,0 +1,33 @@
+# Copyright 2023 The Cornac Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+import random
+import time
+import unittest
+
+from cornac.datasets import gowalla
+
+
+class TestGowalla(unittest.TestCase):
+
+    def test_load_checkins(self):
+        random.seed(time.time())
+        if random.random() > 0.8:
+            checkins = gowalla.load_checkins()
+            self.assertEqual(len(checkins), 6442892)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/cornac/datasets/test_tafeng.py

@@ -0,0 +1,33 @@
+# Copyright 2023 The Cornac Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+import random
+import time
+import unittest
+
+from cornac.datasets import tafeng
+
+
+class TestTafeng(unittest.TestCase):
+
+    def test_load_basket(self):
+        random.seed(time.time())
+        if random.random() > 0.8:
+            baskets = tafeng.load_basket()
+            self.assertEqual(len(baskets), 817741)
+
+
+if __name__ == "__main__":
+    unittest.main()

tests/cornac/datasets/test_yoochoose.py

@@ -0,0 +1,38 @@
+# Copyright 2023 The Cornac Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+
+import random
+import time
+import unittest
+
+from cornac.datasets import yoochoose
+
+
+class TestYooChoose(unittest.TestCase):
+
+    def test_load_buy_click_test(self):
+        random.seed(time.time())
+        if random.random() > 0.8:
+            buy = yoochoose.load_buy()
+            click = yoochoose.load_click()
+            test = yoochoose.load_test()
+
+            self.assertEqual(len(buy), 1150753)
+            self.assertEqual(len(click), 33003944)
+            self.assertEqual(len(test), 8251791)
+
+
+if __name__ == "__main__":
+    unittest.main()