add basic test using pytest