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

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

चरण

• 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