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")
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', ))
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',))
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",))
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", ))