progress

Author: flexatone

src/lib.rs

@@ -431,5 +431,6 @@ fn arrayredox(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(first_true_1d_e, m)?)?;
     // m.add_function(wrap_pyfunction!(first_true_1d_g, m)?)?;
     m.add_function(wrap_pyfunction!(first_true_1d, m)?)?;
+    m.add_function(wrap_pyfunction!(first_true_2d, m)?)?;
     Ok(())
 }

tests/test_first_true_2d.py

@@ -0,0 +1,18 @@
+from arrayredox import first_true_2d
+import numpy as np
+
+import pytest
+
+#--------------------------------------------------------------------------
+def test_first_true_2d_a() -> None:
+    a1 = np.arange(20).reshape(4, 5) % 3 == 0
+    pos2 = first_true_2d(a1, 1)
+    assert pos2.tolist() == [0, 1, 2, 0]
+
+    pos1 = first_true_2d(a1, 0)
+    import ipdb; ipdb.set_trace()
+    assert pos1.tolist() == [0, 1, 2, 0, 1]
+
+
+
+