Move class tests under class

@rafaelfranca