add unit tests for expression functions (#1121)

* Adding unit tests for expression functions

* flatten is returning [] not None in CI

* minor unit test change

* Updated unit tests per review suggestions

Author: timsaucer