Removed unused import pytest from test_demo.py