Added tests for DFT