def test_load_data(self): for data_file in self.data_files: data = Data() data.path = data_file['filename'] data.genfromtxt_args['delimiter'] = data_file['delimiter'] if data_file['headers']: data.genfromtxt_args['skip_header'] = 1 yield assert_allclose, data.load_data(), self.raw_data
def test_load_data_with_scale(self): data_file = self.data_files[0] data = Data() data.path = data_file['filename'] data.genfromtxt_args['delimiter'] = data_file['delimiter'] data.scale = (2, 5) raw_data_scaled = [ [ 2 * x for x in self.raw_data[0] ], [ 5 * x for x in self.raw_data[1] ], ] assert_allclose(data.load_data(), raw_data_scaled)