#!/usr/bin/env python3from unittest import TestCase, main
from sympy import pprint, Matrix, Identity, ZeroMatrix
print('2.')
deff(n):
returnMatrix([[g(n, i, j)for j inrange(1, n + 1)]
for i inrange(1, n + 1)])
defg(n, i, j):
if i % n == j - 1:
return 1
return 0
defh(n):
returnMatrix([[k(i, j) for j inrange(n)]
for i inrange(n)])
defk(i, j):
if i == j:
return 1
return 0
for o in [f(3), f(4)]:
pprint(o)
print()
classMyTestCase(TestCase):
deftest3(self):
self.assertEqual(f(3) ** 3, h(3))
deftest4(self):
self.assertEqual(f(4) ** 4, h(4))
if __name__ == '__main__':
main()
0 コメント:
コメントを投稿