fix #79

fplll · Jul 13, 2017 · 2bc30f5 · 2bc30f5
1 parent 04d1f73
commit 2bc30f5
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/src/fpylll/fplll/integer_matrix.pyx b/src/fpylll/fplll/integer_matrix.pyx
@@ -1432,6 +1432,18 @@ cdef class IntegerMatrix:
     def from_file(cls, filename):
         """Construct new matrix from file.
 
+        >>> import tempfile
+        >>> A = IntegerMatrix.random(10, "qary", k=5, bits=20)
+
+        >>> fn = tempfile.mktemp()
+        >>> fh = open(fn, "w")
+        >>> fh.write(str(A))
+        >>> fh.close()
+
+        >>> B = IntegerMatrix.from_file(fn)
+        >>> A == B
+        True
+
         :param filename: name of file to read from
 
         """
@@ -1455,7 +1467,7 @@ cdef class IntegerMatrix:
                     raise RuntimeError("Integer type '%s' not understood."%(<IntegerMatrix>A)._type)
 
                 for j, v in enumerate(values):
-                    A._set(i, j, v)
+                    (<IntegerMatrix>A)._set(i, j, v)
         return A