#! /usr/bin/python3

class Squares():
    """
    Use a list comprehension to square the elements of a list.
    """

    def __init__(self, alist):
        self.list = [x**2 for x in alist]

def test_list_comprehension()
    alist = [1, 2, 3]
    test_squares = Squares(alist)
    assert test_squares[0] == 1
    assert test_squares[1] == 4
    assert test_squares[2] == 9