Exemple #1
0
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
Exemple #2
0
    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