Fix and test stream_count

nilmdb/client/client.py

@@ -184,7 +184,8 @@ class Client(object):
         Return the number of rows of data in the stream that satisfy
         the given timestamps.
         """
-        return int(self.stream_extract(path, start, end, count = True)[0])
+        counts = list(self.stream_extract(path, start, end, count = True))
+        return int(counts[0])
 
 class StreamInserter(object):
     """Object returned by stream_insert_context() that manages

tests/test_client.py

@@ -251,6 +251,9 @@ class TestClient(object):
                 x = client.stream_intervals("/newton/prep")
         in_("nesting calls is not supported", str(e.exception))
 
+        # Test count
+        eq_(client.stream_count("/newton/prep"), 14400)
+
         client.close()
 
     def test_client_06_generators(self):