如何针对使用 Matplotlib 的代码编写单元测试?
要针对代码编写单元测试用例,我们可以考虑将数组作为x点并将其绘制为y=x^2的图。在测试中,我们将提取y_data为X数据points.-
脚步
创建一个方法,即plot_sqr_curve(x)使用plot()方法绘制x和x^2并返回绘图。
要进行测试,请使用unittest.TestCase。
test_curve_sqr_plot()包含以下语句的Write方法。
为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