def test_multi_arg(self): def fake_slice(records, i, j): return records[i:j] a_cls = common.partial_append_action(fake_slice, ['i', 'j']) a = a_cls([], 'functions') a(None, self.namespace, [0, 2]) f = self.namespace.functions[0] self.assertEqual([0, 1], f(range(10)))
def test_no_arg(self): def head(records): return records[:2] a_cls = common.partial_append_action(head) a = a_cls([], 'functions') a(None, self.namespace, None) f = self.namespace.functions[0] self.assertEqual([0, 1], f(range(10)))
def test_multi_arg(self): def fake_slice(records, i, j): return records[i:j] a_cls = common.partial_append_action(fake_slice, ['i', 'j']) a = a_cls([], 'functions') a(None, self.namespace, [0, 2]) f = self.namespace.functions[0] self.assertEqual([0, 1], f(list(range(10))))
def test_no_arg(self): def head(records): return records[:2] a_cls = common.partial_append_action(head) a = a_cls([], 'functions') a(None, self.namespace, None) f = self.namespace.functions[0] self.assertEqual([0, 1], f(list(range(10))))
def test_single_arg(self): def head(records, n): return records[:n] a_cls = common.partial_append_action(head, 'n') a = a_cls([], 'functions') a(None, self.namespace, 2) f = self.namespace.functions[0] self.assertEqual([0, 1], f(list(range(10))))