コード例 #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")
コード例 #2
0
ファイル: test_pipeline.py プロジェクト: zztt/zipline-chinese
    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', ))
コード例 #3
0
ファイル: test_pipeline.py プロジェクト: AtwooTM/zipline
    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',))
コード例 #4
0
ファイル: test_pipeline.py プロジェクト: testmana2/zipline
    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",))
コード例 #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", ))