def setUpClass(self):
self.folder = dcs.get_tests_dir()
self.contents = 'Hello, World!\n'
self.filepath = os.path.join(self.folder, 'temp_file.txt')
self.badpath = r'AA:\non_existent_path\bad_file.txt'
with open(self.filepath, 'wt') as file:
file.write(self.contents)
def setUp(self):
self.folder = dcs.get_root_dir()
self.text = 'from .bpe import'
self.text2 = 'from .bpe import'
self.folder2 = dcs.get_tests_dir()
self.filepath = os.path.join(self.folder2, 'temp_file.py')
self.filename = os.path.join(self.folder2, '__init__2.py')
def test_saving_plot(self):
this_filename = os.path.join(
dcs.get_tests_dir(), self.opts.case_name + ' - Figure Title.png')
self.opts.save_plot = True
dcs.setup_plots(self.fig, self.opts)
# remove file
os.remove(this_filename)
def setUp(self):
self.fold1 = dcs.get_tests_dir()
self.file1 = os.path.join(self.fold1, 'temp_file.pyc')
self.fold2 = os.path.join(self.fold1, 'temp_sub_dir')
self.file2 = os.path.join(self.fold2, 'temp_file2.pyc')
dcs.write_text_file(self.file1, 'Text.')
os.makedirs(self.fold2)
dcs.write_text_file(self.file2, 'More text.')
def test_show_link(self):
this_filename = os.path.join(dcs.get_tests_dir(), self.opts.case_name + ' - Figure Title.png')
self.opts.save_plot = True
self.opts.show_link = True
with dcs.capture_output() as out:
dcs.setup_plots(self.fig, self.opts)
output = out.getvalue().strip()
out.close()
# remove file
os.remove(this_filename)
self.assertTrue(output.startswith('Plots saved to <a href="'))
def test_show_link(self):
this_filename = os.path.join(
dcs.get_tests_dir(), self.opts.case_name + ' - Figure Title.png')
self.opts.save_plot = True
self.opts.show_link = True
with dcs.capture_output() as out:
dcs.setup_plots(self.fig, self.opts)
output = out.getvalue().strip()
out.close()
# remove file
os.remove(this_filename)
self.assertTrue(output.startswith('Plots saved to <a href="'))
def setUpClass(cls):
cls.folder = dcs.get_tests_dir()
file1 = os.path.join(cls.folder, 'temp_code_01.py')
file2 = os.path.join(cls.folder, 'temp_code_02.py')
file3 = os.path.join(cls.folder, 'temp_code_03.m')
cont1 = 'Line 1\n\nAnother line\n Line with leading spaces\n'
cont2 = '\n\n Start line\nNo Bad tab lines\n Start and end line \nAnother line\n\n'
cont3 = '\n\n Start line\n\tBad tab line\n Start and end line \nAnother line\n\n'
cls.files = [file1, file2, file3]
dcs.write_text_file(file1, cont1)
dcs.write_text_file(file2, cont2)
dcs.write_text_file(file3, cont3)
cls.bad1 = " Line 004: '\\tBad tab line\\n'"
cls.bad2 = " Line 005: ' Start and end line \\n'"
def setUp(self):
self.fig = plt.figure()
self.fig.canvas.set_window_title('Figure Title')
ax = self.fig.add_subplot(111)
x = np.arange(0, 10, 0.1)
y = np.sin(x)
ax.plot(x, y)
ax.set_title('X vs Y')
ax.set_xlabel('time [years]')
ax.set_ylabel('value [radians]')
self.opts = dcs.Opts()
self.opts.case_name = 'Testing'
self.opts.show_plot = True
self.opts.save_plot = False
self.opts.save_path = dcs.get_tests_dir()
def setUp(self):
self.fig = plt.figure()
self.fig.canvas.set_window_title('Figure Title')
ax = self.fig.add_subplot(111)
x = np.arange(0, 10, 0.1)
y = np.sin(x)
ax.plot(x, y)
ax.set_title('X vs Y')
ax.set_xlabel('time [years]')
ax.set_ylabel('value [radians]')
self.opts = dcs.Opts()
self.opts.case_name = 'Testing'
self.opts.show_plot = True
self.opts.save_plot = False
self.opts.save_path = dcs.get_tests_dir()
def setUp(self):
self.folder = dcs.get_tests_dir()
self.old_name = 'test1'
self.new_name = 'test2'
self.print_status = True
self.old_dir = os.path.join(self.folder, self.old_name)
self.new_dir = os.path.join(self.folder, self.new_name)
self.git_dir = os.path.join(self.old_dir, '.git')
self.files = ['__init__.py', '__init__.pyc', '__init__.misc']
# make some files
if not os.path.isdir(self.old_dir):
os.mkdir(self.old_dir)
if not os.path.isdir(self.git_dir):
os.mkdir(self.git_dir)
dcs.write_text_file(os.path.join(self.old_dir, '__init__.py'),'# Init file for "temp1".\n')
dcs.write_text_file(os.path.join(self.old_dir, '__init__.pyc'),'')
dcs.write_text_file(os.path.join(self.old_dir, '__init__.misc'),'# Misc file for "temp1".\n')
def setUpClass(cls):
# create data
cls.time = np.arange(0, 10, 0.1)
cls.data = np.sin(cls.time)
cls.title = 'Test Plot'
cls.folder = dcs.get_tests_dir()
cls.plot_type = ['png', 'jpg']
# create the figure and set the title
cls.fig = plt.figure()
cls.fig.canvas.set_window_title(cls.title)
# add an axis and plot the data
ax = cls.fig.add_subplot(111)
ax.plot(cls.time, cls.data)
# add labels and legends
ax.set_xlabel('Time [year]')
ax.set_ylabel('Value [units]')
ax.set_title(cls.title)
# show a grid
ax.grid(True)
def setUpClass(cls):
# create data
cls.time = np.arange(0, 10, 0.1)
cls.data = np.sin(cls.time)
cls.title = 'Test Plot'
cls.folder = dcs.get_tests_dir()
cls.plot_type = ['png', 'jpg']
# create the figure and set the title
cls.fig = plt.figure()
cls.fig.canvas.set_window_title(cls.title)
# add an axis and plot the data
ax = cls.fig.add_subplot(111)
ax.plot(cls.time, cls.data)
# add labels and legends
ax.set_xlabel('Time [year]')
ax.set_ylabel('Value [units]')
ax.set_title(cls.title)
# show a grid
ax.grid(True)
def setUpClass(self):
# create data
self.time = np.arange(0, 10, 0.1)
self.data = np.sin(self.time)
self.title = 'Test Plot'
self.folder = dcs.get_tests_dir()
self.plot_type = ['png', 'jpg']
# turn interaction off to make the plots draw all at once on a show() command
plt.ioff()
# create the figure and set the title
self.fig = plt.figure()
self.fig.canvas.set_window_title(self.title)
# add an axis and plot the data
ax = self.fig.add_subplot(111)
ax.plot(self.time, self.data)
# add labels and legends
plt.xlabel('Time [year]')
plt.ylabel('Value [units]')
plt.title(self.title)
# show a grid
plt.grid(True)
# show the plot
plt.show(block=False)
def setUp(self):
folder = dcs.get_tests_dir()
self.results1 = _Example_SaveAndLoad()
self.results2 = _Example_SaveAndLoadPickle()
self.save_path1 = os.path.join(folder, 'results_test_save.hdf5')
self.save_path2 = os.path.join(folder, 'results_test_save.pkl')
def setUpClass(cls):
cls.folder = dcs.get_tests_dir()
cls.contents = 'Hello, World!\n'
cls.filepath = os.path.join(cls.folder, 'temp_file.txt')
cls.badpath = r'AA:\non_existent_path\bad_file.txt'
def test_saving(self):
self.logger.set_level(0)
self.opti_opts.max_iters = 0
self.opti_opts.output_folder = dcs.get_tests_dir()
self.opti_opts.output_results = 'temp_results.hdf5'
dcs.run_bpe(self.opti_opts)
def setUp(self):
self.bpe_results = dcs.BpeResults()
self.bpe_results.num_evals = 5
self.filename = os.path.join(dcs.get_tests_dir(), 'test_bpe_results.hdf5')
self.filename2 = self.filename.replace('hdf5', 'pkl')
dcs.Logger.set_level(1)
def setUp(self):
self.folder = os.path.join(dcs.get_tests_dir(), 'temp_dir')
self.subdir = os.path.join(dcs.get_tests_dir(), 'temp_dir', 'temp_dir2')
self.filename = os.path.join(self.folder, 'temp_file.txt')
self.subfile = os.path.join(self.subdir, 'temp_file.txt')
self.text = 'Hello, World!\n'
def test_function(self):
folder = dcs.get_tests_dir()
self.assertEqual(folder, os.path.join(dcs.get_root_dir(), 'tests'))
def test_saving_plot(self):
this_filename = os.path.join(dcs.get_tests_dir(), self.opts.case_name + ' - Figure Title.png')
self.opts.save_plot = True
dcs.setup_plots(self.fig, self.opts)
# remove file
os.remove(this_filename)
def setUp(self):
folder = dcs.get_tests_dir()
self.results1 = _Example_SaveAndLoad()
self.results2 = _Example_SaveAndLoadPickle()
self.save_path1 = os.path.join(folder, 'results_test_save.hdf5')
self.save_path2 = os.path.join(folder, 'results_test_save.pkl')