Matplotlibを使用するコードに対して単体テストを作成するにはどうすればよいですか？

コードに対して単体テストケースを作成するために、配列をxポイントとして取り、それを y =x^2としてプロットするプロットを検討できます。 テスト中に、 y_dataを抽出します xデータポイントの場合。−

ステップ

• メソッドを作成します。つまり、 plot_sqr_curve（x） plot（）を使用してxとx^2をプロットするには メソッドとプロットを返します。
• テストするには、unittest.TestCaseを使用します。
• test_curve_sqr_plot（）を記述します 次のステートメントを含むメソッド。
  • 曲線をプロットするためにxのデータポイントを作成します。
  • 上記のx個のデータポイントを使用して、y個のデータポイントを作成します。
  • xおよびyデータポイントを使用して、曲線をプロットします。
  • pt（ステップ5から）を使用して、xおよびyデータを抽出します。
  • 指定された式が真であるかどうかを確認します。

例

import unittest
import numpy as np
from matplotlib import pyplot as plt
def plot_sqr_curve(x):
   """
   Plotting x points with y = x^2.
   """
return plt.plot(x, np.square(x))
class TestSqrCurve(unittest.TestCase):
def test_curve_sqr_plot(self):
   x = np.array([1, 3, 4])
   y = np.square(x)
   pt, = plot_sqr_curve(x)
   y_data = pt.get_data()[1]
   x_data = pt.get_data()[0]
   self.assertTrue((y == y_data).all())
   self.assertTrue((x == x_data).all())
if __name__ == '__main__':
   unittest.main()

出力

Ran 1 test in 1.587s
OK