Source code for runlmc.linalg.identity

# Copyright (c) 2016, Vladimir Feinberg
# Licensed under the BSD 3-clause license (see LICENSE)

import numpy as np

from .matrix import Matrix
from ..util.docs import inherit_doc

# TODO(test)
[docs]@inherit_doc
class Identity(Matrix):
    def __init__(self, n):
        super().__init__(n, n)

[docs]    def matvec(self, x):
        return x

[docs]    def matmat(self, x):
        return x

[docs]    def as_numpy(self):
        return np.identity(self.shape[0])

[docs]    def upper_eig_bound(self):
        return 1