add tests for package main