Update test_cell.py