Fix typing for get_cross_sections
and add tests
#2246
Merged
get_cross_sections
and add tests
#2246