Add proper support for constants

Author: james-d-mitchell

libsemigroups_pybind11/__init__.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 
-# Copyright (c) 2021, J. D. Mitchell
+# Copyright (c) 2021-2022, J. D. Mitchell
 #
 # Distributed under the terms of the GPL license version 3.
 #
@@ -9,11 +9,12 @@
 # pylint: disable=no-name-in-module
 
 """
-This package provides a the user-facing python part of libsemigroups_pybind11
+This package provides the user-facing python part of libsemigroups_pybind11
 """
 
 from _libsemigroups_pybind11 import (
     PBR,
+    NEGATIVE_INFINITY,
     POSITIVE_INFINITY,
     UNDEFINED,
     ActionDigraph,

libsemigroups_pybind11/matrix.py

@@ -55,25 +55,34 @@ class MatrixKind(Enum):
 }
 
 
+def _convert_matrix_args(*args):
+    if not (len(args) == 1 and isinstance(*args, list)):
+        return args
+    # Convert POSITIVE_INFINITY and NEGATIVE_INFINITY to integers
+    return (
+        [[z if isinstance(z, int) else z.to_int() for z in y] for y in args[0]],
+    )
+
+
 def Matrix(kind: MatrixKind, *args):
     """
     Constructs a matrix, basically just delegates to
     _libsemigroups_pybind11
     """
     if not isinstance(kind, MatrixKind):
         raise TypeError("the 1st argument must be a MatrixKind")
-    return _Matrix[kind](*args)
+    return _Matrix[kind](*_convert_matrix_args(*args))
 
 
 def make_identity(kind: MatrixKind, *args) -> Matrix:
     """
     Construct the identity matrix of the appropriate type.
     """
-    return _Matrix[kind].make_identity(*args)
+    return _Matrix[kind].make_identity(*_convert_matrix_args(*args))
 
 
 def make(kind: MatrixKind, *args) -> Matrix:
     """
     Construct a matrix of the appropriate type.
     """
-    return _Matrix[kind].make(*args)
+    return _Matrix[kind].make(*_convert_matrix_args(*args))