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.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(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)