[Add] Tests: type counts

Signed-off-by: oleg komarov <o.komarov11@imperial.ac.uk>