Added set val functionality and Intermission

Csv_F.py

@@ -53,21 +53,33 @@ class Csv_F:
 		return self.__csv_data[r_index][c_index]
 
 	def set_row(self, id, new_row, is_header=False, force=False):
-		data_len = self.__get_width()
+		index = self.get_index(id, is_header, is_row=True)
+		data_len = self.get_width()
 		new_data_len = len(new_row)
+
 		if new_data_len != data_len and not force:
 			raise Exception('New data length does not match target. Set force flag if required.') from None
 			return False
 		else:
-			constraints = ((new_row, data_len) if new_data_len >= data_len else (self.get_row(id, is_header), new_data_len))
-			for i, j in enumerate(constraints[0]):
-				return None
+			# (new_row if new_data_len >= data_len else new_row[0:data_len])
+			if new_data_len >= data_len:
+				for data, col in enumerate(self.__csv_data[index]):
+					
+			else:
+				for data, col in enumerate(new_row):				
 
 	def set_col(self, id, new_col, is_header=False, force=False):
 		return None
 
 	def set_val(self, new_val, r_id, c_id, r_is_header=False, c_is_header=False):
-		
+		if type(new_val) in [int, float, str, bool]:
+			r_index = self.get_index(r_id, r_is_header, is_row=True)
+			c_index = self.get_index(c_id, c_is_header)
+			self.__csv_data[r_index][c_index] = new_val
+			return True
+		else:
+			raise Exception('The value entered must be a singlton. ' +str(type(new_val)) +' Not accepted.') from None
+			return False
 
 	def get_row_headers(self):
 		return self.__row_headers

User_File.py

@@ -54,5 +54,35 @@ class CSV_Testing(unittest.TestCase):
 		self.assertRaises(Exception, CSV2.get_val, 'R2', 'NA', r_is_header=True, c_is_header=True)
 		self.assertRaises(Exception, CSV2.get_val, 'NA', 'C1', r_is_header=True, c_is_header=True)
 
+	def test_val_setting(self):
+		CSV1 = Csv_Man.load('test1.csv', has_r_headers=True, has_c_headers=True)
+		CSV2 = Csv_Man.load('test2.csv', has_r_headers=True, has_c_headers=True, r_delim=';', c_delim=':')
+
+		self.assertEqual(CSV1.get_val(0, 0), '1')
+		CSV1.set_val('Q', 0, 0)
+		self.assertEqual(CSV1.get_val(0, 0), 'Q')
+
+		self.assertEqual(CSV2.get_val('R1', 'C3', r_is_header=True, c_is_header=True), '3')
+		CSV2.set_val('Q', 'R1', 'C3', r_is_header=True, c_is_header=True)
+		self.assertEqual(CSV2.get_val('R1', 'C3', r_is_header=True, c_is_header=True), 'Q')
+
+	def test_val_error(self):
+		CSV1 = Csv_Man.load('test1.csv', has_r_headers=True, has_c_headers=True)
+		CSV2 = Csv_Man.load('test2.csv', has_r_headers=True, has_c_headers=True, r_delim=';', c_delim=':')
+
+		self.assertEqual(CSV1.get_val(0, 0), '1')
+		self.assertRaises(Exception, CSV1.set_val, ['Q'], 0, 0)
+		self.assertEqual(CSV1.get_val(0, 0), '1')
+
+		self.assertEqual(CSV2.get_val('R1', 'C3', r_is_header=True, c_is_header=True), '3')
+		self.assertRaises(Exception, CSV1.set_val, ['Q'], 'R1', 'C3', r_is_header=True, c_is_header=True)
+		self.assertEqual(CSV2.get_val('R1', 'C3', r_is_header=True, c_is_header=True), '3')
+
+	def test_row_set(self):
+		CSV1 = Csv_Man.load('test1.csv', has_r_headers=True, has_c_headers=True)
+		CSV2 = Csv_Man.load('test2.csv', has_r_headers=True, has_c_headers=True, r_delim=';', c_delim=':')
+
+		CSV1.set_row(0, ['Q','Q', 'Q'])
+
 if __name__ == '__main__':
 	unittest.main(verbosity=1)