if __name__ == "__main__": # TODO(ruilongli): write tests pass