Working on RML module.

1 files changed, 13 insertions, 8 deletions

diff --git a/Software/Visual_Studio/Tango.UnitTesting/ColorLib_TST.cs b/Software/Visual_Studio/Tango.UnitTesting/ColorLib_TST.cs
index dc19c77dd..b2e32afce 100644
--- a/Software/Visual_Studio/Tango.UnitTesting/ColorLib_TST.cs
+++ b/Software/Visual_Studio/Tango.UnitTesting/ColorLib_TST.cs
@@ -14,6 +14,7 @@ using System.Diagnostics;
 using Tango.Documents;
 using Tango.Core.IO;
 using Tango.MachineStudio.ColorLab.ViewModels;
+using Tango.BL.Calibration;
 
 namespace Tango.UnitTesting
 {
@@ -167,25 +168,29 @@ namespace Tango.UnitTesting
         }
 
         [TestMethod]
-        public void Write_Calibration_Data_Excel_File()
+        public void Read_Write_Calibration_Data_Excel_File()
         {
             String temp_file = Helper.GetTempFolderPath() + "\\CalibrationData.xlsx";
-            File.Copy(Helper.GetResourcePath("Calibration Data Template.xlsx"), temp_file);
 
-            CalibrationDataVM vm = new CalibrationDataVM();
+            List<CalibrationPoint> w_calibration_points = new List<CalibrationPoint>();
 
             for (int i = 0; i < 10; i++)
             {
-                vm.CalibrationPoints.Add(new CalibrationDataPointVM()
+                w_calibration_points.Add(new CalibrationPoint()
                 {
-                     X = i + 1,
-                     Y = (i + 1) * 10,
+                    X = i + 1,
+                    Y = (i + 1) * 10,
                 });
             }
 
-            using (ExcelWriter writer = new ExcelWriter(temp_file))
+            CalibrationHelper.ExportCalibrationDataToExcel(w_calibration_points, temp_file);
+            var r_calibration_points = CalibrationHelper.ImportCalibrationDataFromExcel(temp_file);
+
+            Assert.AreEqual(w_calibration_points.Count, r_calibration_points.Count, "Read calibration points count does not match the written calibration points count.");
+
+            for (int i = 0; i < w_calibration_points.Count; i++)
             {
-                writer.WriteData(vm.CalibrationPoints, "CalibrationData");
+                Assert.AreEqual(w_calibration_points[i].X + w_calibration_points[i].Y, r_calibration_points[i].X + r_calibration_points[i].Y, $"Calibration points at index {i} are not equal.");
             }
 
             Helper.ShowInExplorer(temp_file);