Want to create org docs with embedded java blocks which will generate testing data

I have seen http://ehneilsen.net/notebook/orgExamples/org-examples.html#sec-19 

but have not been able to find an example where there are any import statements e.g. 

import java.time.*;

Can anyone tell me how I might achieve this?

Thanks,

Tim