Chapter 16: テスト

pytestを使って、2乗を計算する関数 square(n) のテストを書いてください。入力3で結果9を期待するテストケースを含めてください。