#!/usr/bin/python3
"""Unittests for base
"""

import unittest
from models.square import Square


class TestSquare(unittest.TestCase):
    """Define unit test for Square model"""

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)

    def test_initialization_without_arguments(self):

        self.assertRaises(TypeError, Square)

if __name__ == '__main__':
    unittest.main()

34db9c0e6d29b4a386147816d57c7ff4193ec080
<mask token>


class TestSquare(unittest.TestCase):
    <mask token>

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)
    <mask token>


<mask token>

<mask token>


class TestSquare(unittest.TestCase):
    <mask token>

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)

    def test_initialization_without_arguments(self):
        self.assertRaises(TypeError, Square)


<mask token>

<mask token>


class TestSquare(unittest.TestCase):
    """Define unit test for Square model"""

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)

    def test_initialization_without_arguments(self):
        self.assertRaises(TypeError, Square)


if __name__ == '__main__':
    unittest.main()

<mask token>
import unittest
from models.square import Square


class TestSquare(unittest.TestCase):
    """Define unit test for Square model"""

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)

    def test_initialization_without_arguments(self):
        self.assertRaises(TypeError, Square)


if __name__ == '__main__':
    unittest.main()

#!/usr/bin/python3
"""Unittests for base
"""

import unittest
from models.square import Square


class TestSquare(unittest.TestCase):
    """Define unit test for Square model"""

    def test_initialization_success(self):
        s1 = Square(5)
        s2 = Square(10)
        self.assertEqual(s1.id, 5)
        self.assertEqual(s2.id, 6)

    def test_initialization_without_arguments(self):

        self.assertRaises(TypeError, Square)

if __name__ == '__main__':
    unittest.main()
