cols = columns should be changed to columns = columns in the DFMock constructor.

[Hellisotherpeople]

Your initial example didn't work and I had to look at source code to fix it.

Luckily, DFMock worked for me right after I made that change.