Fix error in Count iterator

.github/workflows/build_docs.yaml

@@ -16,7 +16,7 @@ jobs:
       with:
         python-version: '3.11'
     - name: Install dependencies and local packages
-      run: python -m pip install sphinx pydata-sphinx-theme sphinx-autoapi sphinx-design
+      run: python -m pip install sphinx pydata-sphinx-theme "sphinx-autoapi != 3.4.0" sphinx-design
     - name: Build HTML documentation with Sphinx
       run: |
         make html

.github/workflows/check_docs.yaml

@@ -14,7 +14,7 @@ jobs:
       with:
         python-version: '3.11'
     - name: Install dependencies and local packages
-      run: python -m pip install sphinx pydata-sphinx-theme sphinx-autoapi sphinx-design
+      run: python -m pip install sphinx pydata-sphinx-theme "sphinx-autoapi != 3.4.0" sphinx-design
     - name: Build HTML documentation with Sphinx
       run: |
         make html

src/tempe/data_view.py

@@ -270,7 +270,7 @@ def __iter__(self):
         i = self.start
         while True:
             yield i
-            i += 1
+            i += self.step
 
     def __len__(self):
         return None

tests/tempe/test_data_views.py

@@ -0,0 +1,56 @@
+# SPDX-FileCopyrightText: 2024-present Unital Software <[email protected]>
+#
+# SPDX-License-Identifier: MIT
+
+import unittest
+
+from tempe.data_view import Count
+
+
+class TestCount(unittest.TestCase):
+
+    def test_count(self):
+        """Test count starts and steps as expected."""
+
+        count = Count(10, 5)
+        count_iter = iter(count)
+
+        self.assertEqual(next(count_iter), 10)
+        self.assertEqual(next(count_iter), 15)
+        self.assertEqual(next(count_iter), 20)
+
+    def test_count_getitem(self):
+        """Test count getitem works as expected."""
+
+        count = Count(10, 5)
+
+        self.assertEqual(count[0], 10)
+        self.assertEqual(count[1], 15)
+        self.assertEqual(count[2], 20)
+
+    def test_count_default(self):
+        """Test count default starts and steps as expected."""
+
+        count = Count()
+        count_iter = iter(count)
+
+        self.assertEqual(next(count_iter), 0)
+        self.assertEqual(next(count_iter), 1)
+        self.assertEqual(next(count_iter), 2)
+
+    def test_count_default_getitem(self):
+        """Test count default getitem works as expected."""
+
+        count = Count()
+
+        self.assertEqual(count[0], 0)
+        self.assertEqual(count[1], 1)
+        self.assertEqual(count[2], 2)
+
+
+if __name__ == "__main__":
+    result = unittest.main()
+    if not result.wasSuccessful():
+        import sys
+
+        sys.exit(1)