def test_wrong_file_number_in_filebase(self):
writer_config = {
'options': {
'filebase': '/tmp/%m/%Y-some_folder_{file_number}/{groups[1]}_',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
writer.close()
with self.assertRaisesRegexp(KeyError, 'filebase option should not contain'):
writer.create_filebase_name(('g1', 'g2'), file_name='filename')
def test_wrong_file_number_in_filebase(self):
writer_config = {
'options': {
'filebase':
'/tmp/%m/%Y-some_folder_{file_number}/{groups[1]}_',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
writer.close()
with self.assertRaisesRegexp(KeyError,
'filebase option should not contain'):
writer.create_filebase_name(('g1', 'g2'), file_name='filename')
def test_create_filebase_name(self):
writer_config = {
'options': {
'filebase': '/tmp/%m/%Y-some_folder_{groups[0]}/{groups[1]}_{file_number}_',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
writer.close()
date = datetime.datetime.now()
expected = (date.strftime('/tmp/%m/%Y-some_folder_g1'), 'filename')
self.assertEqual(writer.create_filebase_name(('g1', 'g2'), file_name='filename'), expected)
def test_get_file_number_not_implemented(self):
writer_config = {
'options': {
'filebase': '/tmp/',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
self.assertIsInstance(writer.get_file_suffix('', ''), basestring)
path, file_name = writer.create_filebase_name([])
self.assertEqual(path, '/tmp')
writer.close()
def test_get_file_number_not_implemented(self):
writer_config = {
'options': {
'filebase': '/tmp/',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
self.assertIsInstance(writer.get_file_suffix('', ''), basestring)
path, file_name = writer.create_filebase_name([])
self.assertEqual(path, '/tmp')
writer.close()
def test_create_filebase_name(self):
writer_config = {
'options': {
'filebase':
'/tmp/%m/%Y-some_folder_{groups[0]}/{groups[1]}_{file_number}_',
}
}
writer = FilebaseBaseWriter(writer_config, meta())
writer.close()
date = datetime.datetime.now()
expected = (date.strftime('/tmp/%m/%Y-some_folder_g1'), 'filename')
self.assertEqual(
writer.create_filebase_name(('g1', 'g2'), file_name='filename'),
expected)