def test_save_data_input_text(self):
a = InputFileLoaderCheckerSaver()
a._input_text= "hello"
a.save_data_to_file=True
a._save_data()
assert_equal(self.tempdir.read(
('out0002','out0002_input_original.py'), 'utf-8').strip().splitlines(),
'hello'.splitlines())
def test_determine_output_stem_directory(self):
a = InputFileLoaderCheckerSaver()
a.directory = os.path.join(self.tempdir.path, 'what')
a._determine_output_stem()
assert_equal(
a._file_stem,
os.path.join(self.tempdir.path, 'what', 'out0002', 'out0002'))
def test_check_zero_or_all(self):
a = InputFileLoaderCheckerSaver()
a.a=4
a.b=6
a.c=None
a._zero_or_all = ['a b c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_save_data_grid_data_dicts(self):
a = InputFileLoaderCheckerSaver()
a._grid_data_dicts= {'data': np.arange(6).reshape(3,2)}
a.save_data_to_file=True
a._save_data()
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002.csv')))
def test_attribute_defaults(self):
a = InputFileLoaderCheckerSaver()
a._input_text = 'b=6'
a._attributes = "a b".split()
a._attribute_defaults = {'a': 24}
a._initialize_attributes()
a._transfer_attributes_from_inputfile()
assert_equal(a.a, 24)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'b=6')
def test_attribute_defaults(self):
a = InputFileLoaderCheckerSaver()
a._input_text = 'b=6'
a._attributes = "a b".split()
a._attribute_defaults = {'a': 24}
a._initialize_attributes()
a._transfer_attributes_from_inputfile()
assert_equal(a.a, 24)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'b=6')
def test_check_attributes_to_force_same_len(self):
a = InputFileLoaderCheckerSaver()
a.a = [4, 5]
a.c = None
a._attributes_to_force_same_len = ['a c'.split()]
a.check_input_attributes()
assert_equal(a.c, [None, None])
def test_check_attributes_that_should_be_lists(self):
a = InputFileLoaderCheckerSaver()
a.a = 4
a.b = 6
a._attributes_that_should_be_lists = ['b']
a.check_input_attributes()
assert_equal(a.a, 4)
assert_equal(a.b, [6])
def test_init_from_str(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b".split()
a._initialize_attributes()
a.__init__('a=4\nb=6')
assert_equal(a.a, 4)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'a=4\nb=6')
def test_save_data_input_ext(self):
a = InputFileLoaderCheckerSaver()
a._input_text= "hello"
a.input_ext= '.txt'
a.save_data_to_file=True
a._save_data()
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002_input_original.txt')))
def test_init_from_fileobj(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b".split()
a._initialize_attributes()
with open(os.path.join(self.tempdir.path, 'inp1.py'), 'r') as f:
a.__init__(f)
assert_equal(a.a, 4)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'a=4\nb=6')
def test_save_data_grid_data_dicts_data_ext(self):
a = InputFileLoaderCheckerSaver()
a._grid_data_dicts= {'data': np.arange(6).reshape(3,2)}
a.save_data_to_file=True
a.data_ext = ".txt"
a._save_data()
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002.txt')))
def test_save_data_parsed(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b ".split()
a.save_data_to_file=True
# a._initialize_attributes()
a.a=4
a.b=6
a._save_data()
# print(os.listdir(self.tempdir.path))
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
assert_equal(self.tempdir.read(
('out0002','out0002_input_parsed.py'), 'utf-8').strip().splitlines(),
'a = 4\nb = 6'.splitlines())
def test_check_attributes_to_force_same_len(self):
a = InputFileLoaderCheckerSaver()
a.a = [4,5]
a.c=None
a._attributes_to_force_same_len = ['a c'.split()]
a.check_input_attributes()
assert_equal(a.c, [None, None])
def test_check_zero_or_all(self):
a = InputFileLoaderCheckerSaver()
a.a = 4
a.b = 6
a.c = None
a._zero_or_all = ['a b c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_save_data_input_text(self):
a = InputFileLoaderCheckerSaver()
a._input_text= "hello"
a.save_data_to_file=True
a._save_data()
assert_equal(self.tempdir.read(
('out0002','out0002_input_original.py'), 'utf-8').strip().splitlines(),
'hello'.splitlines())
def test_save_figures_figure_ext(self):
a = InputFileLoaderCheckerSaver()
a.save_figures_to_file=True
a.figure_ext='.pdf'
fig = matplotlib.pyplot.figure()
ax = fig.add_subplot('111')
ax.plot(4,5)
fig.set_label('sing')
a._figures=[fig]
a._save_figures()
a._figures=None
matplotlib.pyplot.clf()
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002_sing.pdf')))
def test_init_from_str(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b".split()
a._initialize_attributes()
a.__init__('a=4\nb=6')
assert_equal(a.a, 4)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'a=4\nb=6')
def test_check_attributes_that_should_be_lists(self):
a = InputFileLoaderCheckerSaver()
a.a=4
a.b=6
a._attributes_that_should_be_lists = ['b']
a.check_input_attributes()
assert_equal(a.a, 4)
assert_equal(a.b, [6])
def test_save_data_input_ext(self):
a = InputFileLoaderCheckerSaver()
a._input_text= "hello"
a.input_ext= '.txt'
a.save_data_to_file=True
a._save_data()
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002_input_original.txt')))
def test_init_from_fileobj(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b".split()
a._initialize_attributes()
with open(os.path.join(self.tempdir.path, 'inp1.py'), 'r') as f:
a.__init__(f)
assert_equal(a.a, 4)
assert_equal(a.b, 6)
assert_equal(a._input_text, 'a=4\nb=6')
def test_save_data_parsed(self):
a = InputFileLoaderCheckerSaver()
a._attributes = "a b ".split()
a.save_data_to_file=True
# a._initialize_attributes()
a.a=4
a.b=6
a._save_data()
# print(os.listdir(self.tempdir.path))
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
assert_equal(self.tempdir.read(
('out0002','out0002_input_parsed.py'), 'utf-8').strip().splitlines(),
'a = 4\nb = 6'.splitlines())
def test_save_figures_figure_ext(self):
a = InputFileLoaderCheckerSaver()
a.save_figures_to_file=True
a.figure_ext='.pdf'
fig = matplotlib.pyplot.figure()
ax = fig.add_subplot('111')
ax.plot(4,5)
fig.set_label('sing')
a._figures=[fig]
a._save_figures()
a._figures=None
matplotlib.pyplot.clf()
# print(os.listdir(os.path.join(self.tempdir.path,'out0002')))
ok_(os.path.isfile(os.path.join(
self.tempdir.path, 'out0002','out0002_sing.pdf')))
def test_check_attributes_that_should_have_same_len_pairs(self):
a = InputFileLoaderCheckerSaver()
a.a = [2, 3]
a.c = [3]
a._attributes_that_should_have_same_len_pairs = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_check_attributes_that_should_have_same_x_limits(self):
a = InputFileLoaderCheckerSaver()
a.a = PolyLine([0, 1], [2, 5])
a.c = PolyLine([0, 7], [5, 6])
a._attributes_that_should_have_same_x_limits = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_determine_output_stem_create_directory(self):
a = InputFileLoaderCheckerSaver()
a.create_directory = False
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0002')
def test_determine_output_stem_create_directory(self):
a = InputFileLoaderCheckerSaver()
a.create_directory = False
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0002')
def test_check_attributes_that_should_have_same_x_limits(self):
a = InputFileLoaderCheckerSaver()
a.a = PolyLine([0,1], [2,5])
a.c = PolyLine([0,7], [5,6])
a._attributes_that_should_have_same_x_limits = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_check_attributes_that_should_have_same_len_pairs(self):
a = InputFileLoaderCheckerSaver()
a.a = [2, 3]
a.c = [3]
a._attributes_that_should_have_same_len_pairs = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_determine_output_stem_defaults(self):
a = InputFileLoaderCheckerSaver()
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0002\\out0002')
def test_determine_output_stem_prefix(self):
a = InputFileLoaderCheckerSaver()
a.prefix = 'hello_'
a._determine_output_stem()
assert_equal(a._file_stem, '.\\hello_0001\\hello_0001')
def test_determine_output_stem_defaults(self):
a = InputFileLoaderCheckerSaver()
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0002\\out0002')
def test_determine_output_stem_overwrite(self):
a = InputFileLoaderCheckerSaver()
a.overwrite = True
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0001\\out0001')
def test_check_one_implies_others(self):
a = InputFileLoaderCheckerSaver()
a.a = 4
a.c=None
a._one_implies_others = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_determine_output_stem_prefix(self):
a = InputFileLoaderCheckerSaver()
a.prefix = 'hello_'
a._determine_output_stem()
assert_equal(a._file_stem, '.\\hello_0001\\hello_0001')
def test_check_at_least_one(self):
a = InputFileLoaderCheckerSaver()
a.c=None
a._at_least_one = ['c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_check_at_least_one(self):
a = InputFileLoaderCheckerSaver()
a.c = None
a._at_least_one = ['c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_check_one_implies_others(self):
a = InputFileLoaderCheckerSaver()
a.a = 4
a.c = None
a._one_implies_others = ['a c'.split()]
assert_raises(ValueError, a.check_input_attributes)
def test_determine_output_stem_directory(self):
a = InputFileLoaderCheckerSaver()
a.directory = os.path.join(self.tempdir.path, 'what')
a._determine_output_stem()
assert_equal(a._file_stem, os.path.join(self.tempdir.path,'what', 'out0002', 'out0002'))
def test_determine_output_stem_overwrite(self):
a = InputFileLoaderCheckerSaver()
a.overwrite = True
a._determine_output_stem()
assert_equal(a._file_stem, '.\\out0001\\out0001')
