Updated tests and use of num_cols

Author: kanhereashwin

gnss_lib_py/navdata/navdata.py

@@ -898,7 +898,7 @@ def __next__(self):
         x_curr : gnss_lib_py.navdata.navdata.NavData
             Current column (based on iteration count)
         """
-        if self.curr_col >= self.num_cols:
+        if self.curr_col >= len(self):
             raise StopIteration
         x_curr = self.copy(rows=None, cols=self.curr_col)
         self.curr_col += 1

tests/navdata/test_navdata.py

@@ -176,12 +176,28 @@ def test_init_np(numpy_array):
         data = NavData(numpy_array=pd.DataFrame([0]))
 
 
-def test_len_num_cols_rows(data):
+def test_len_rows(data):
+    """Test that `len()` and `rows` return correct output.
+
+    Parameters
+    ----------
+    data : gnss_lib_py.navdata.navdata.NavData
+        Simple version of NavData to use for test.
+    """
     assert len(data) == 6
-    assert data.num_cols == len(data)
     assert data.rows == ['names', 'integers', 'floats', 'strings']
 
 
+def test_num_cols(data):
+    """Test that `num_cols` returns correct output.
+
+    Parameters
+    ----------
+    data : gnss_lib_py.navdata.navdata.NavData
+        Simple version of NavData to use for test."""
+    assert data.num_cols == 6
+
+
 
 @pytest.mark.parametrize('pandas_df',
                         [