Pythonでndarrayのテストがしたい．

Question

前提・実現したいこと

Pythonにおいてndarray形式の出力をするhogehoge関数のtestをしたい．

発生している問題・エラーメッセージ

  np.testing.assert_array_equal(a, [[-2.10531461e+00 -3.98845323e-01 -4.36655961e-02  1.25119167e+00
                                                                                                     ^
SyntaxError: invalid syntax

ndarray形式のままコピペしたからか上のようなメッセージが出てくる(','区切りにするか，np.testing.assert_array_equal()以外の方法でする必要がある？)

該当のソースコード

a = {hogehoge関数からndarray形式で出力}
b = {他のファイルでhogehoge関数で出力したndarray形式のデータをコピペしたもの}

import numpy as np
np.testing.assert_array_equal(A, B)

試したこと

コピペするデータを

np.set_printoptions(threshold=np.inf)
print(hogehoge(), sep=',')

で','区切りにしようとしたが，','区切りにもならない．

補足情報（FW/ツールのバージョンなど）

Python3.6.5

Answer 1

printによる文字列変換はstr()と同様に行われます。これは "略式の" つまり読み易い object の文字列表現を返すもので、オブジェクトの値を再現可能な表現とは異なります。

printの際に明示的にreprを使うことで望みのものが得られると思います。

>>> import numpy as np
>>> a = np.arange(5)
>>> print(a)
[0 1 2 3 4]
>>> print(str(a))
[0 1 2 3 4]
>>> print(repr(a))
array([0, 1, 2, 3, 4])  # 先頭にnp.をつければそのまま文字列として評価できる

2. 組み込み関数 — Python 3.6.5 ドキュメント | print
4. 組み込み型 — Python 3.6.5 ドキュメント | str
2. 組み込み関数 — Python 3.6.5 ドキュメント | repr

ただし、コードの目的や意図によっては元のnumpy配列オブジェクトをnp.save等でダンプしてそれを読み込むようにしたほうがスマートかもしれません。

Comments

[退会済みユーザー]

ありがとうございます．おっしゃる通りに試すとうまく","で区切ることができました．

Answer 2

テストをするために、正解データとして一度なんらかのnp.ndarrayを出力し、その内容をコピぺし、hogehoge関数の結果と比較することでtestとしようという意図ですよね。

コピペしやすいように出力する方法としては、こちらのようにできると思いますが、
単体テストと考えると、それほど複雑なデータを評価しなくても、いくつかの
例をコードに埋め込み、試験した方が分かりやすいと思います。