def test_load_file_default(self): """Test default load for text file and h5 file without extension.""" filename = self.current_dir + '/test_data' self.assertRaises(IOError, load_file, filename) filename = self.current_dir + '/test_data2' default = load_file(filename) self.assertEqual(default, ["some text"])
def test_load_file_json(self): json_dict = load_file(self.current_dir+'/test_data.json') self.assertEqual(json_dict, {'x': [0, 1]})
def test_filetype_option(self): filename = self.current_dir + '/test_data' hdf = load_file(filename, filetype="hdf") self.assertIsInstance(hdf, FileHDFio) self.assertEqual(hdf['content/key'], 'value')
def test_load_file_ProjectHDF(self): pr = Project(self.current_dir + '/test_pr') pr_hdf = load_file(self.current_dir + '/test_data.h5', project=pr) self.assertIsInstance(pr_hdf, ProjectHDFio) self.assertEqual(pr_hdf['content/key'], 'value') pr.remove(enable=True)
def test_load_file_hdf(self): hdf = load_file(self.current_dir+'/test_data.h5') self.assertIsInstance(hdf, FileHDFio) self.assertEqual(hdf['content/key'], 'value')
def test_load_file_csv(self): csv = load_file(self.current_dir+'/test_data.csv') content = pd.read_csv(self.current_dir+'/test_data.csv') self.assertTrue(content.equals(csv))
def test_load_file_stream_txt(self): with open(self.current_dir + '/test_data.txt') as f: content = f.readlines() with open(self.current_dir + '/test_data.txt') as f: txt = load_file(f) self.assertEqual(txt, content)