working on moving research code as unit tests