コード例 #1
0
    def test_out_of_order(self):
        """
        Tests the case where arguments are not in the correct order.
        """
        def f(a, b):
            pass

        with pytest.raises(MismatchedArguments):
            verify_callable_argspec(f, [Argument("b"), Argument("a")])
コード例 #2
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_out_of_order(self):
        """
        Tests the case where arguments are not in the correct order.
        """
        def f(a, b):
            pass

        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(f, [Argument('b'), Argument('a')])
コード例 #3
0
    def test_arg_subset(self):
        """
        Tests when the args are a subset of the expectations.
        """
        def f(a, b):
            pass

        with pytest.raises(NotEnoughArguments):
            verify_callable_argspec(
                f, [Argument("a"), Argument("b"),
                    Argument("c")])
コード例 #4
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_arg_subset(self):
        """
        Tests when the args are a subset of the expectations.
        """
        def f(a, b):
            pass

        with self.assertRaises(NotEnoughArguments):
            verify_callable_argspec(
                f, [Argument('a'), Argument('b'),
                    Argument('c')])
コード例 #5
0
    def test_bound_method(self):
        class C(object):
            def f(self, a, b):
                pass

        method = C().f

        verify_callable_argspec(method, [Argument("a"), Argument("b")])
        with self.assertRaises(NotEnoughArguments):
            # Assert that we don't count self.
            verify_callable_argspec(
                method,
                [Argument("self"),
                 Argument("a"), Argument("b")],
            )
コード例 #6
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_bound_method(self):
        class C:
            def f(self, a, b):
                pass

        method = C().f

        verify_callable_argspec(method, [Argument('a'), Argument('b')])
        with self.assertRaises(NotEnoughArguments):
            # Assert that we don't count self.
            verify_callable_argspec(
                method,
                [Argument('self'),
                 Argument('a'), Argument('b')],
            )
コード例 #7
0
ファイル: events.py プロジェクト: MridulS/zipline
    def __new__(cls, rule=None, callback=None, check_args=True):
        callback = callback or (lambda *args, **kwargs: None)
        if check_args:
            # Check the callback provided.
            verify_callable_argspec(callback, [
                Argument('context' if check_args else Argument.ignore),
                Argument('data' if check_args else Argument.ignore)
            ])

            # Make sure that the rule's should_trigger is valid. This will
            # catch potential errors much more quickly and give a more helpful
            # error.
            verify_callable_argspec(getattr(rule, 'should_trigger'),
                                    [Argument('dt')])

        return super(cls, cls).__new__(cls, rule=rule, callback=callback)
コード例 #8
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_ignore_default(self):
        """
        Tests that ignoring defaults works as intended.
        """
        def f(a=1):
            pass

        verify_callable_argspec(f, [Argument('a')])
コード例 #9
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_default(self):
        """
        Tests when an argument expects a default and it is present.
        """
        def f(a=1):
            pass

        verify_callable_argspec(f, [Argument('a', 1)])
コード例 #10
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_no_default(self):
        """
        Tests when an argument expects a default and it is not present.
        """
        def f(a):
            pass

        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(f, [Argument('a', 1)])
コード例 #11
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_wrong_default(self):
        """
        Tests the case where a default is expected, but the default provided
        does not match the one expected.
        """
        def f(a=1):
            pass

        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(f, [Argument('a', 2)])
コード例 #12
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_ignore_name(self):
        """
        Tests ignoring a param name.
        """
        def f(a):
            pass

        def g(b):
            pass

        def h(c=1):
            pass

        expected_args = [Argument(Argument.ignore, Argument.no_default)]
        verify_callable_argspec(f, expected_args)
        verify_callable_argspec(f, expected_args)
        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(h, expected_args)
コード例 #13
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_any_default(self):
        """
        Tests the any_default option.
        """
        def f(a=1):
            pass

        def g(a=2):
            pass

        def h(a):
            pass

        expected_args = [Argument('a', Argument.any_default)]
        verify_callable_argspec(f, expected_args)
        verify_callable_argspec(g, expected_args)
        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(h, expected_args)
コード例 #14
0
    def test_arg_superset(self):
        def f(a, b, c):
            pass

        with pytest.raises(TooManyArguments):
            verify_callable_argspec(f, [Argument("a"), Argument("b")])
コード例 #15
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_mismatched_args(self):
        def f(a, b):
            pass

        with self.assertRaises(MismatchedArguments):
            verify_callable_argspec(f, [Argument('c'), Argument('d')])
コード例 #16
0
    def test_mismatched_args(self):
        def f(a, b):
            pass

        with pytest.raises(MismatchedArguments):
            verify_callable_argspec(f, [Argument("c"), Argument("d")])
コード例 #17
0
ファイル: test_argcheck.py プロジェクト: fangchenli/zipline
    def test_arg_superset(self):
        def f(a, b, c):
            pass

        with self.assertRaises(TooManyArguments):
            verify_callable_argspec(f, [Argument('a'), Argument('b')])