import os
import shutil
from fooof import FOOOF, FOOOFGroup
from fooof.sim import gen_group_power_spectra

# Simulate
freqs, powers = gen_group_power_spectra(10, (1, 50), [-2, 0], [10, 5, 5])

# Fit
fm = FOOOF(verbose=False)
fm.fit(freqs, powers[0])

fg = FOOOFGroup(verbose=False)
fg.fit(freqs, powers)

# Make a tmp directory
if os.path.isdir('tmpdir_'):
    shutil.rmtree('tmpdir_')
os.mkdir('tmpdir_')

# Relative Path
fm.plot(file_name='tmpdir_/plot')
fm.save(file_name='tmpdir_/results')
fm.save_report(file_name='tmpdir_/report')

fg.plot(file_name='tmpdir_/fg_plot')
fg.save(file_name='tmpdir_/fg_results', save_results=True)
fg.save_report(file_name='tmpdir_/fg_report')

assert os.path.isfile('tmpdir_/plot.png')
assert os.path.isfile('tmpdir_/results.json')
assert os.path.isfile('tmpdir_/report.pdf')
assert os.path.isfile('tmpdir_/fg_plot.png')
assert os.path.isfile('tmpdir_/fg_results.json')
assert os.path.isfile('tmpdir_/fg_report.pdf')

# Absolute Path
cwd = os.getcwd()

fm.plot(file_name=cwd + '/tmpdir_/plot')
fm.save(file_name=cwd + '/tmpdir_/results')
fm.save_report(file_name=cwd + '/tmpdir_/report')

fg.plot(file_name=cwd + '/tmpdir_/fg_plot')
fg.save(file_name=cwd + '/tmpdir_/fg_results', save_results=True)
fg.save_report(file_name=cwd + '/tmpdir_/fg_report')

assert os.path.isfile(cwd + '/tmpdir_/plot.png')
assert os.path.isfile(cwd + '/tmpdir_/results.json')
assert os.path.isfile(cwd + '/tmpdir_/report.pdf')
assert os.path.isfile(cwd + '/tmpdir_/fg_plot.png')
assert os.path.isfile(cwd + '/tmpdir_/fg_results.json')
assert os.path.isfile(cwd + '/tmpdir_/fg_report.pdf')

shutil.rmtree('tmpdir_')