Fix spacing error
Grab test data from pytest
Start adding some tests