test_app.py

"""Tests for the flask app"""
import unittest
from app import app

class TestSquare(unittest.TestCase):
    """Tests for the square function"""
    def test_positive(self):
        """Test for a positive number"""
        with app.test_client() as test_client:
            response = test_client.get('/square/2')
            self.assertTrue(response.status_code == 200)
            self.assertTrue(response.data == b'4')
            self.assertFalse(response.data == b'5')
    def test_negative(self):
        """Test for a negative number"""
        with app.test_client() as test_client:
            response = test_client.get('/square/-2')
            self.assertTrue(response.status_code == 200)
            self.assertTrue(response.data == b'4')
            self.assertFalse(response.data == b'-4')
    def test_zero(self):
        """Test for zero"""
        with app.test_client() as test_client:
            response = test_client.get('/square/0')
            self.assertTrue(response.status_code == 200)
            self.assertTrue(response.data == b'0')
            self.assertFalse(response.data == b'1')
if __name__ == '__main__':
    unittest.main()