fix: Added backend handling to the module tests (#26988)

we were only testing with the `numpy` backend uptil now