Python - 数学 - ベクトル - ドット積の練習

行列プログラマー
原著

開発環境

• macOS Sierra - Apple (OS)
• Emacs (Text Editor)
• Python 3.6 (プログラミング言語)

行列プログラマー(Philip N. Klein (著)、 松田 晃一 (翻訳)、 弓林 司 (翻訳)、 脇本 佑紀 (翻訳)、 中田 洋 (翻訳)、 齋藤 大吾 (翻訳)、オライリージャパン)の2章(ベクトル)、2.14(問題)、ドット積の練習、問題 2.14.9 を取り組んでみる。

問題 2.14.9

コード(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)})


class Test(unittest.TestCase):

    def setUp(self):
        pass

    def setDown(self):
        pass

    def test_a(self):
        v = list2vec([1, 0]) * list2vec([5, 4321])
        self.assertEqual(v, 5)

    def test_b(self):
        v = list2vec([0, 1]) * list2vec([12345, 6])
        self.assertEqual(v, 6)

    def test_c(self):
        v = list2vec([-1, 3]) * list2vec([5, 7])
        self.assertEqual(v, 16)

    def test_d(self):
        v = list2vec([-math.sqrt(2) / 2, math.sqrt(2) / 2]) * \
            list2vec([math.sqrt(2) / 2, -math.sqrt(2) / 2])
        self.assertEqual(v, -1)

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

入出力結果(Terminal, IPython)

$ ./sample14_9.py -v
test_a (__main__.Test) ... ok
test_b (__main__.Test) ... ok
test_c (__main__.Test) ... ok
test_d (__main__.Test) ... FAIL

======================================================================
FAIL: test_d (__main__.Test)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "./sample14_9.py", line 36, in test_d
    self.assertEqual(v, -1)
AssertionError: -1.0000000000000002 != -1

----------------------------------------------------------------------
Ran 4 tests in 0.001s

FAILED (failures=1)
$