Fix assumed shape character arrays

Author: rjfarmer

gfort2py/module_parse.py

@@ -196,7 +196,10 @@ def dtype(self):
             elif k == 8:
                 return np.cdouble
         elif t == "CHARACTER":
-            return f"S{self.strlen.value}"
+            try:
+                return f"S{self.strlen.value}"
+            except AttributeError:
+                return "S"
 
         raise NotImplementedError(f"Object of type {t} and kind {k} not supported yet")
 

tests/strings.f90

@@ -357,4 +357,16 @@ subroutine set_chr_star_star(x)
 	end subroutine set_chr_star_star
 
 
+	subroutine check_assumed_shape_str(x)
+		character(len=*), dimension(:), intent(in) :: x
+		integer  :: i
+
+
+		do i=1,ubound(x,dim=1)
+			write(*,*) trim(x(i))
+		end do
+
+
+	end subroutine check_assumed_shape_str
+
 end module strings

tests/strings_test.py

@@ -289,3 +289,10 @@ def test_set_str_array_dt_out(self):
     def test_set_chr_star_star(self):
         res = x.set_chr_star_star("            ")
         assert res.args["x"] == "abcdefghijkl"
+
+    def test_check_assumed_shape_str(self, capfd):
+        y = np.array(["a/b/c/d/e/f/g"], dtype="S")
+
+        res = x.check_assumed_shape_str(y)
+        out, err = capfd.readouterr()
+        assert out.strip() == "a/b/c/d/e/f/g"