Complete the test
Fix includes
Fix test case
Ensure in_dbn mode for each rbm