示例#1
0
class RectangleTestCase(TestCase):
    def setUp(self):
        self.rectangle = Rectangle(width=7, height=8)

    def test_rectangle_area(self):
        """
        Test that we can calculate the area of a rectangle
        """
        area = self.rectangle.area()

        self.assertEqual(area, 56)

    @mock.patch('shapes.tweet')
    def test_rectangle_broadcast(self, mock_tweet):
        """
        Tests that we call tweet with a formatted message
        """
        self.rectangle.broadcast()

        mock_tweet.assert_called_with('My rectangle is 7 by 8')
示例#2
0
class RectangleTestCase(TestCase):

    def setUp(self):
        self.rectangle = Rectangle(width=7, height=8)

    def test_rectangle_area(self):
        """
        Test that we can calculate the area of a rectangle
        """
        area = self.rectangle.area()

        self.assertEqual(area, 56)

    @mock.patch('shapes.tweet')
    def test_rectangle_broadcast(self, mock_tweet):
        """
        Tests that we call tweet with a formatted message
        """
        self.rectangle.broadcast()

        mock_tweet.assert_called_with('My rectangle is 7 by 8')