मैं Matplotlib का उपयोग करने वाले कोड के विरुद्ध यूनिट परीक्षण कैसे लिख सकता हूं?

एक कोड के खिलाफ यूनिट टेस्ट केस लिखने के लिए, हम एक प्लॉट पर विचार कर सकते हैं जो एक सरणी को x पॉइंट के रूप में लेता है और इसे y=x^2 के रूप में प्लॉट करता है। परीक्षण के दौरान, हम y_data निकालेंगे x डेटा बिंदुओं के लिए।−

कदम

• एक विधि बनाएं, यानी, plot_sqr_curve(x) प्लॉट () . का उपयोग करके x और x^2 को प्लॉट करने के लिए विधि और प्लॉट वापस करें।
• परीक्षण करने के लिए, unittest.TestCase. . का उपयोग करें
• लिखें test_curve_sqr_plot() विधि जिसमें निम्नलिखित कथन शामिल हैं।
  • वक्र को प्लॉट करने के लिए x के लिए डेटा बिंदु बनाएं।
  • उपरोक्त x डेटा बिंदुओं का उपयोग करके, y डेटा बिंदु बनाएं।
  • x और y डेटा बिंदुओं का उपयोग करके, वक्र प्लॉट करें।
  • पीटी का उपयोग करके (चरण 5 से), एक्स और वाई डेटा निकालें।
  • जांचें कि दिया गया व्यंजक सत्य है या नहीं।

उदाहरण

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