refactor tests slightly
Add test using mock and real dataframes / series.