Python: Building battleship test.