Làm cách nào để viết các bài kiểm tra đơn vị dựa trên mã sử dụng Matplotlib?

Để viết các trường hợp kiểm thử đơn vị dựa trên mã, chúng ta có thể coi một biểu đồ nhận một mảng là x điểm và vẽ biểu đồ đó là y =x ^ 2. Trong khi thử nghiệm, chúng tôi sẽ trích xuất y_data cho x điểm dữ liệu.−

Các bước

• Tạo một phương thức, tức là plot_sqr_curve (x) để vẽ biểu đồ x và x ^ 2 bằng cách sử dụng plot () và trả về cốt truyện.
• Để kiểm tra, hãy sử dụng unittest.TestCase.
• Viết test_curve_sqr_plot () phương thức bao gồm các câu lệnh sau.
  • Tạo các điểm dữ liệu cho x để vẽ đường cong.
  • Sử dụng x điểm dữ liệu ở trên, tạo điểm dữ liệu y.
  • Sử dụng các điểm dữ liệu x và y, vẽ đường cong.
  • Sử dụng pt (từ bước 5), trích xuất dữ liệu x và y.
  • Kiểm tra xem biểu thức đã cho có đúng hay không.

Ví dụ

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()

Đầu ra

Ran 1 test in 1.587s
OK