Update test_layout.py