Restore test_controller

I thought all the tests in here were things related specific to the old rust controller because the first few were. But there are real tensor engine tests in here. So I put them back

Differential Revision: [D87955377](https://our.internmc.facebook.com/intern/diff/D87955377/)

ghstack-source-id: 325854076
Pull Request resolved: #2013

Author: zdevito