Restore test_controller (#2013)

Summary:
Pull Request resolved: #2013

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
ghstack-source-id: 327431136
exported-using-ghexport

Reviewed By: mariusae, colin2328

Differential Revision: D87955377

fbshipit-source-id: 0d8ea6762c7b4829e666e1eaa5984a0326c11ac3

Author: zdevito