__all__ = [
    'load_sample_data', 'eq', 'aa_eq', 'duplicate_traj', 'Timer', 'tempfolder',
    'amberhome', 'cpptraj_test_dir', 'cpptraj_dat_dir', 'get_fn', 'get_remd_fn',
    'assert_equal_topology'
]

if cpptrajhome:
    cpptraj_test_dir = os.path.join(cpptrajhome, 'test')
    cpptraj_dat_dir = os.path.join(cpptrajhome, 'dat')
else:
    if amberhome:
        cpptraj_test_dir = os.path.join(amberhome, 'AmberTools', 'src',
                                        'cpptraj', 'test')
        cpptraj_dat_dir = os.path.join(amberhome, 'dat')
    else:
        cpptrajhome = ''
        amberhome = ''
        cpptraj_test_dir = ''
        cpptraj_dat_dir = ''

DEFAULT_PATH = os.path.dirname(__file__) + "/../../cpptraj"

if os.path.exists(DEFAULT_PATH):
    cpptraj_test_dir = os.path.abspath(os.path.join(DEFAULT_PATH, 'test'))
    cpptraj_dat_dir = os.path.abspath(os.path.join(DEFAULT_PATH, 'dat'))

if __name__ == "__main__":
    print(amberhome)
    print(cpptraj_test_dir)
    print(cpptraj_dat_dir)