2017年5月6日土曜日

開発環境

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の3章(ベクトル空間)、3.1(線形結合)、3.1.3(係数から線形結合へ)のクイズ 3.1.7を取り組んでみる。

クイズ 3.1.7

コード(Emacs)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import unittest
import math
from vec import Vec


def list2vec(l):
    return Vec(set(range(len(l))), {i: v for i, v in enumerate(l)})


def lin_comb(vlist, clist):
    return sum([c * v for c, v in zip(clist, vlist)])


class Test(unittest.TestCase):

    def setUp(self):
        pass

    def setDown(self):
        pass

    def test_0(self):
        self.assertEqual(lin_comb([], []), 0)

    def test_1(self):
        vlist = [list2vec([0, 1])]
        clist = [5]
        self.assertEqual(lin_comb(vlist, clist),
                         list2vec([0, 5]))

    def test_2(self):
        vlist = [list2vec([0, 1]),
                 list2vec([2, 3])]
        clist = [5, 10]
        self.assertEqual(lin_comb(vlist, clist),
                         list2vec([20, 35]))


if __name__ == '__main__':
    unittest.main()

入出力結果(Terminal, IPython)

$ ./sample1_7.py -v
test_0 (__main__.Test) ... ok
test_1 (__main__.Test) ... ok
test_2 (__main__.Test) ... ok

----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK
$

0 コメント:

コメントを投稿