add other public attributes to gather class

Author: dpgraham4401

src/core/bindings.cpp

@@ -39,5 +39,8 @@ PYBIND11_MODULE(_num, m) {
 PYBIND11_MODULE(_gather, m) {
   py::class_<Gather>(m, "Gather")
       .def(py::init<>())
-      .def_readwrite("id", &Gather::id);
+      .def_readwrite("id", &Gather::id)
+      .def_readwrite("dt", &Gather::dt)
+      .def_readwrite("nt", &Gather::nt)
+      .def_readwrite("nx", &Gather::nx);
 }

src/core/gather/gather.hpp

@@ -8,6 +8,9 @@
 class Gather {
 public:
   int id;
+  int nt;
+  int nx;
+  double dt;
 };
 
 #endif // GATHER_H

tests/libseis/test_add.py

@@ -1,6 +1,4 @@
-import pytest
-
-from libseis import Gather, add, add_one
+from libseis import add, add_one
 
 
 class TestLibseis:
@@ -17,22 +15,3 @@ def test_add(self):
     def test_add_one(self):
         expected = 4
         assert add_one(3) == expected
-
-
-class TestSeismicGather:
-    """Test the Gather class from."""
-
-    @pytest.fixture
-    def gather(self):
-        """Fixture to create a Gather instance."""
-        return Gather()
-
-    def test_gather_init(self, gather):
-        """Test the Gather class base constructor."""
-        assert isinstance(gather, Gather)
-
-    def test_gather_id_is_publicly_accessible(self, gather):
-        """Test that the id attribute is publicly accessible."""
-        my_id = 42
-        gather.id = my_id
-        assert gather.id == my_id

tests/libseis/test_gather.py

@@ -0,0 +1,23 @@
+import pytest
+
+from libseis import Gather
+
+
+class TestSeismicGather:
+    """Test the Gather class from."""
+
+    @pytest.fixture
+    def gather(self):
+        """Fixture to create a Gather instance."""
+        return Gather()
+
+    def test_gather_init(self, gather):
+        """Test the Gather class base constructor."""
+        assert isinstance(gather, Gather)
+
+    @pytest.mark.parametrize("attr", ["id", "dt", "nx", "nt"])
+    def test_public_int_attributes(self, gather, attr):
+        """Test that the id attribute is publicly accessible."""
+        val = 42
+        setattr(gather, attr, val)
+        assert getattr(gather, attr) == val