def test_cli(self, table_9_content):
csv_filepath = os.path.join(self.current_tmp, 'tab.csv')
hepdata_converter._main(['--output-format', 'csv', '--table', 'Table 9', '--separator', ';', self.submission_filepath,
csv_filepath])
with open(csv_filepath, 'r') as csv_file:
self.assertEqual(table_9_content, csv_file.read())
def test_single_file_output(self, oldhepdata_path, oldhepdata_file,
oldhepdata_yaml_file):
data = hepdata_converter.convert(oldhepdata_file,
options={
'input_format':
'oldhepdata',
'output_format':
'yaml',
'single_file':
True,
'hepdata_doi':
'10.17182/hepdata.62535.v1'
})
yaml_single_output_path = os.path.join(self.current_tmp, 'output.yaml')
hepdata_converter._main([
'--input-format', 'oldhepdata', '--output-format', 'yaml',
'--single-file', oldhepdata_path, yaml_single_output_path
])
print('OUTPUT PATH IS {}'.format(yaml_single_output_path))
self.assertTrue(os.path.exists(yaml_single_output_path))
self.assertEqual(oldhepdata_yaml_file, data)
def test_cli(self, oldhepdata_path, oldhepdata_yaml_path):
hepdata_converter._main([
'--input-format', 'oldhepdata', '--output-format', 'yaml',
oldhepdata_path, self.current_tmp
])
self.assertDirsEqual(oldhepdata_yaml_path, self.current_tmp)
def test_diroutput(self, oldhepdata_path, csv_path):
hepdata_converter._main([
'-i', 'oldhepdata', '-o', 'csv', oldhepdata_path, self.current_tmp
])
self.assertDirsEqual(self.current_tmp,
csv_path,
file_content_parser=lambda x: x)
def test_cli(self, table_9_content):
csv_filepath = os.path.join(self.current_tmp, 'tab.csv')
hepdata_converter._main([
'--output-format', 'csv', '--table', 'Table 9', '--separator', ';',
self.submission_filepath, csv_filepath
])
with open(csv_filepath, 'r') as csv_file:
self.assertEqual(table_9_content, csv_file.read())
def test_no_dir_output(self, table_1_content, table_9_content):
csv_filepath = os.path.join(self.current_tmp, 'csv_dir')
hepdata_converter._main(['--output-format', 'csv', '--separator', ',', self.submission_filepath,
csv_filepath])
self.assertTrue(os.path.exists(csv_filepath))
csv_1 = os.path.join(csv_filepath, 'Table1.csv')
self.assertTrue(os.path.exists(csv_1))
with open(csv_1, 'r') as csv_file:
self.assertMultiLineAlmostEqual(table_1_content.strip(), csv_file.read().strip())
csv_2 = os.path.join(csv_filepath, 'Table9.csv')
self.assertTrue(os.path.exists(csv_2))
with open(csv_2, 'r') as csv_file:
self.assertMultiLineAlmostEqual(table_9_content.strip(), csv_file.read().strip())
def test_single_file_output(self, oldhepdata_path, oldhepdata_file, oldhepdata_yaml_file):
data = hepdata_converter.convert(oldhepdata_file,
options={
'input_format': 'oldhepdata',
'output_format': 'yaml',
'single_file': True})
yaml_single_output_path = os.path.join(self.current_tmp, 'output.yaml')
hepdata_converter._main(['--input-format', 'oldhepdata', '--output-format', 'yaml', '--single-file',
oldhepdata_path, yaml_single_output_path])
print('OUTPUT PATH IS {}'.format(yaml_single_output_path))
self.assertTrue(os.path.exists(yaml_single_output_path))
self.assertEqual(oldhepdata_yaml_file, data)
def test_convert_yaml2csv(self, submission_file, table_csv):
output_path = os.path.join(self.current_tmp, 'output.csv')
code, message = hepdata_converter._main(['--input-format', 'yaml', '--output-format', 'csv',
'--table', 'Table 1', '--pack', submission_file, output_path])
self.assertEqual(code, 0, message)
self.assertTrue(os.path.exists(output_path))
with open(output_path) as f:
self.assertMultiLineAlmostEqual(table_csv, f.read())
def test_no_dir_output(self, table_1_content, table_9_content):
csv_filepath = os.path.join(self.current_tmp, 'csv_dir')
hepdata_converter._main([
'--output-format', 'csv', '--validator-schema-version', '0.1.0',
'--separator', ',', self.submission_filepath, csv_filepath
])
self.assertTrue(os.path.exists(csv_filepath))
csv_1 = os.path.join(csv_filepath, 'Table1.csv')
self.assertTrue(os.path.exists(csv_1))
with open(csv_1, 'r') as csv_file:
self.assertMultiLineAlmostEqual(table_1_content.strip(),
csv_file.read().strip())
csv_2 = os.path.join(csv_filepath, 'Table9.csv')
self.assertTrue(os.path.exists(csv_2))
with open(csv_2, 'r') as csv_file:
self.assertMultiLineAlmostEqual(table_9_content.strip(),
csv_file.read().strip())
def test_convert_yaml2csv(self, submission_file, table_csv):
output_path = os.path.join(self.current_tmp, 'output.csv')
code, message = hepdata_converter._main([
'--input-format', 'yaml', '--output-format', 'csv', '--table',
'Table 1', '--pack', submission_file, output_path
])
self.assertEqual(code, 0, message)
self.assertTrue(os.path.exists(output_path))
with open(output_path) as f:
self.assertMultiLineAlmostEqual(table_csv, f.read())
def test_diroutput(self, oldhepdata_path, csv_path):
hepdata_converter._main(['-i', 'oldhepdata',
'-o', 'csv',
oldhepdata_path, self.current_tmp])
self.assertDirsEqual(self.current_tmp, csv_path, file_content_parser=lambda x: x)
def test_version(self):
r, message = hepdata_converter._main(['--version'])
self.assertEqual(r, 0)
self.assertTrue(message.endswith(
hepdata_converter.version.__version__))
def test_convert_yaml2yoda(self):
hepdata_converter._main([
'--input-format', 'yaml', '--output-format', 'csv', '--table',
'Table 1', self.current_tmp,
os.path.join(self.current_tmp, 'output.csv')
])
def test_version(self):
r, message = hepdata_converter._main(['--version'])
self.assertEqual(r, 0)
self.assertTrue(message.endswith(hepdata_converter.version.__version__))
def test_convert_yaml2yoda(self):
hepdata_converter._main(['--input-format', 'yaml', '--output-format', 'csv',
'--table', 'Table 1',
self.current_tmp, os.path.join(self.current_tmp, 'output.csv')])
def test_cli(self, oldhepdata_path, oldhepdata_yaml_path):
hepdata_converter._main(
["--input-format", "oldhepdata", "--output-format", "yaml", oldhepdata_path, self.current_tmp]
)
self.assertDirsEqual(oldhepdata_yaml_path, self.current_tmp)
