Testing environment for the KiT-RT framework https://github.com/CSMMLab/KiT-RT with tensorflow