def test_with_modules(): from cornac.data import TextModule, ImageModule, GraphModule bm = BaseMethod() assert bm.user_text is None assert bm.item_text is None assert bm.user_image is None assert bm.item_image is None assert bm.user_graph is None assert bm.item_graph is None bm.user_text = TextModule() #bm.item_graph = GraphModule() bm._build_modules() try: bm.user_text = ImageModule() except ValueError: assert True #try: # bm.item_text = GraphModule() #except ValueError: # assert True try: bm.user_image = TextModule() except ValueError: assert True #try: # bm.item_image = GraphModule() #except ValueError: # assert True try: bm.user_graph = TextModule() except ValueError: assert True try: bm.item_graph = ImageModule() except ValueError: assert True
def test_with_modalities(self): bm = BaseMethod() self.assertIsNone(bm.user_text) self.assertIsNone(bm.item_text) self.assertIsNone(bm.user_image) self.assertIsNone(bm.item_image) self.assertIsNone(bm.user_graph) self.assertIsNone(bm.item_graph) bm.user_text = TextModality() bm.item_image = ImageModality() bm._build_modalities() try: bm.user_text = ImageModality() except ValueError: assert True try: bm.item_text = ImageModality() except ValueError: assert True try: bm.user_image = TextModality() except ValueError: assert True try: bm.item_image = TextModality() except ValueError: assert True try: bm.user_graph = TextModality() except ValueError: assert True try: bm.item_graph = ImageModality() except ValueError: assert True