026c1ea689b06dc15dfa689dcdb5279f5ecf74e6,tests/links_tests/model_tests/segnet_tests/test_segnet_basic.py,TestSegNetPretrained,test_pretrained,#TestSegNetPretrained#,56

Before Change

    @attr.slow
    def test_pretrained(self):
        SegNetBasic(pretrained_model="camvid")

    @attr.slow
    def test_pretrained_n_class(self):
        SegNetBasic(n_class=11, pretrained_model="camvid")

After Change

            valid = self.n_class is {None, 11}

        if valid:
            SegNetBasic(n_class=self.n_class,
                        pretrained_model=self.pretrained_model)
        else:
            with self.assertRaises(ValueError):
                SegNetBasic(n_class=self.n_class,
                            pretrained_model=self.pretrained_model)

Instances