def test_two_files(self): data = np.arange(6).reshape(3, 2) save_grid_data_to_file([{ 'data': data, 'name': 1 }, { 'data': 2 * data, 'name': 2 }], directory=self.tempdir.path, file_stem="qqq") assert_equal( self.tempdir.read(('qqq', 'qqq1.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines()) assert_equal( self.tempdir.read(('qqq', 'qqq2.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,2 1,4,6 2,8,10""").splitlines())
def test_data_dict_row_labels(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data, 'row_labels':[8,12,6]}, directory=self.tempdir.path) assert_equal(self.tempdir.read(('out_000','out_000.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,item,0,1 0,8,0,1 1,12,2,3 2,6,4,5""").splitlines())
def test_data_dict_column_labels(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data, 'column_labels':['a', 'b']}, directory=self.tempdir.path) assert_equal(self.tempdir.read(('out_000','out_000.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,a,b 0,0,1 1,2,3 2,4,5""").splitlines())
def test_directory(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data}, directory=os.path.join(self.tempdir.path,'g')) assert_equal(self.tempdir.read(('g','out_000','out_000.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines())
def test_data_dict_name(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data, 'name':'xx'}, directory=self.tempdir.path) assert_equal(self.tempdir.read(('out_000','out_000xx.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines())
def test_create_directory(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data}, directory=self.tempdir.path, create_directory=False) assert_equal(self.tempdir.read('out_000.csv', 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines())
def test_file_stem(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data}, directory=self.tempdir.path, file_stem="ppp") assert_equal(self.tempdir.read(('ppp','ppp.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines())
def test_ext(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file({'data': data}, directory=self.tempdir.path, ext=".out") assert_equal(self.tempdir.read(('out_000','out_000.out'), 'utf-8').splitlines(), textwrap.dedent("""\ ,0,1 0,0,1 1,2,3 2,4,5""").splitlines())
def test_two_files(self): data = np.arange(6).reshape(3,2) save_grid_data_to_file([{'data': data, 'name':1}, {'data': 2*data, 'name':2}], directory=self.tempdir.path, file_stem="qqq") assert_equal(self.tempdir.read(('qqq','qqq1.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,1 1,2,3 2,4,5""").splitlines()) assert_equal(self.tempdir.read(('qqq','qqq2.csv'), 'utf-8').splitlines(), textwrap.dedent("""\ idex,0,1 0,0,2 1,4,6 2,8,10""").splitlines())