def SetDelimiter(self, deli):
DelimitedTextReader.SetDelimiter(self, deli)
def SetTimeDelta(self, dt):
DelimitedTextReader.SetTimeDelta(self, dt)
def GetTimestepValues(self):
"""This is critical for registering the timesteps"""
return DelimitedTextReader.GetTimestepValues(self)
def __init__(self):
DelimitedTextReader.__init__(self)
def AddFileName(self, fname):
DelimitedTextReader.AddFileName(self, fname)
def SetComments(self, identifier):
DelimitedTextReader.SetComments(self, identifier)
def test_no_titles(self):
"""`DelimitedTextReader`: file without headers"""
reader = DelimitedTextReader()
reader.AddFileName(self.commafilename)
reader.set_delimiter(',')
reader.set_skip_rows(2)
reader.set_has_titles(False)
reader.set_comments('!')
reader.Update()
table = reader.GetOutput()
self._check_shape(table)
self._check_titles(table, titles=['Field 0', 'Field 1', 'Field 2'])
self._check_array_types(table)
self._check_array_values(table)
return
def set_skip_rows(self, skip):
DelimitedTextReader.set_skip_rows(self, skip)
def set_comments(self, identifier):
DelimitedTextReader.set_comments(self, identifier)
def set_delimiter(self, deli):
DelimitedTextReader.set_delimiter(self, deli)
def set_split_on_white_space(self, flag):
DelimitedTextReader.set_split_on_white_space(self, flag)
def get_time_step_values(self):
"""This is critical for registering the timesteps"""
return DelimitedTextReader.get_time_step_values(self)
def set_time_delta(self, dt):
DelimitedTextReader.set_time_delta(self, dt)
def test_comma_read(self):
"""`DelimitedTextReader`: comma delimited file"""
reader = DelimitedTextReader()
reader.AddFileName(self.commafilename)
reader.set_delimiter(',')
reader.set_skip_rows(1)
reader.set_comments('!')
reader.Update()
table = reader.GetOutput()
self._check_shape(table)
self._check_titles(table)
self._check_array_types(table)
self._check_array_values(table)
return
def SetSplitOnWhiteSpace(self, flag):
DelimitedTextReader.SetSplitOnWhiteSpace(self, flag)
def set_has_titles(self, flag):
DelimitedTextReader.set_has_titles(self, flag)
def SetSkipRows(self, skip):
DelimitedTextReader.SetSkipRows(self, skip)
def SetUseTab(self, flag):
DelimitedTextReader.SetUseTab(self, flag)
def SetHasTitles(self, flag):
DelimitedTextReader.SetHasTitles(self, flag)
def test_tab_read(self):
"""`DelimitedTextReader`: tab delimited file"""
reader = DelimitedTextReader()
reader.AddFileName(self.tabfilename)
reader.set_split_on_white_space(True)
reader.set_skip_rows(1)
reader.set_comments('!')
reader.Update()
table = reader.GetOutput()
self._check_shape(table)
self._check_titles(table)
self._check_array_types(table)
self._check_array_values(table)
return
