How to identify a test Class using AST Parser?

I want to make a tool for detecting Test Smells.For that I need to find out test classes or test Function.

I parse the whole project and check all funtions.