Add dia data layer to benchmarks

[Ericgig]

• Add benchmark for DIA.
• Run steadystate test with CSR instead of DIA.
• Remove the return from benchmark functions. Recent pytest raise a warning.

[Ericgig]

I opened #48 for the steadystate issue and reverted those changes here.