Esempio n. 1
0
    def test_remove(self):
        f = SomeFactor()
        p = Pipeline(columns={"f": f})

        with pytest.raises(KeyError):
            p.remove("not_a_real_name")

        assert f == p.remove("f")

        with pytest.raises(KeyError, match="f"):
            p.remove("f")
Esempio n. 2
0
    def test_remove(self):
        f = SomeFactor()
        p = Pipeline(columns={'f': f})

        with self.assertRaises(KeyError) as e:
            p.remove('not_a_real_name')

        self.assertEqual(f, p.remove('f'))

        with self.assertRaises(KeyError) as e:
            p.remove('f')

        self.assertEqual(e.exception.args, ('f', ))
Esempio n. 3
0
    def test_remove(self):
        f = SomeFactor()
        p = Pipeline(columns={'f': f})

        with self.assertRaises(KeyError) as e:
            p.remove('not_a_real_name')

        self.assertEqual(f, p.remove('f'))

        with self.assertRaises(KeyError) as e:
            p.remove('f')

        self.assertEqual(e.exception.args, ('f',))
Esempio n. 4
0
    def test_remove(self):
        f = SomeFactor()
        p = Pipeline(columns={"f": f})

        with self.assertRaises(KeyError) as e:
            p.remove("not_a_real_name")

        self.assertEqual(f, p.remove("f"))

        with self.assertRaises(KeyError) as e:
            p.remove("f")

        self.assertEqual(e.exception.args, ("f",))
Esempio n. 5
0
    def test_remove(self):
        f = SomeFactor()
        p = Pipeline(columns={"f": f})

        with self.assertRaises(KeyError) as e:
            p.remove("not_a_real_name")

        self.assertEqual(f, p.remove("f"))

        with self.assertRaises(KeyError) as e:
            p.remove("f")

        self.assertEqual(e.exception.args, ("f", ))