Adding tests for correct type hinting for `array` and classes