Updated __main__ unit test
1 file changed