remove code not needed from base.py
add more tests in test_policy