diff options
Diffstat (limited to 'tests/demo_blocks.py')
| -rw-r--r-- | tests/demo_blocks.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/demo_blocks.py b/tests/demo_blocks.py index 9a7a02f..2b2d8dd 100644 --- a/tests/demo_blocks.py +++ b/tests/demo_blocks.py @@ -67,8 +67,9 @@ class TagSource(gras.Block): self._values = values def work(self, ins, outs): - offset = self.nitems_written(0) - self.post_output_tag(0, gras.Tag(offset, 'key', self._values[0])) + offset = self.num_items_produced(0) + tag = gras.Tag(offset, 'key', self._values[0]) + self.post_output_tag(0, tag) self.produce(0, len(outs[0])) self._values = self._values[1:] if not self._values: @@ -86,7 +87,7 @@ class TagSink(gras.Block): return tuple(self._values) def work(self, ins, outs): - max_read = self.nitems_read(0) + len(ins[0]) + max_read = self.num_items_consumed(0) + len(ins[0]) for tag in self.get_input_tags(0): if tag.offset < max_read: self._values.append(tag.value) |