consistency for naming conventions, param orders

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)