Add tests

Author: UsamaEquinorAFK

frontend/src/functions/Formatting.tsx

@@ -69,7 +69,7 @@ export const convertSimulationToChartData = (simulation: SimulationResults, expe
     };
 };
 
-const convertSimulationQueriesResultToChartDataTable = (
+export const convertSimulationQueriesResultToChartDataTable = (
     simulationQueriesResult: UseSimulationQueriesResult
 ): ChartDataTable[] => {
     const chartDataTables: ChartDataTable[] = [];
@@ -92,7 +92,7 @@ const convertSimulationQueriesResultToChartDataTable = (
     return chartDataTables;
 };
 
-const convertExperimentResultsToChartDataTable = (experimentResults: ExperimentResult[]): ChartDataTable[] => {
+export const convertExperimentResultsToChartDataTable = (experimentResults: ExperimentResult[]): ChartDataTable[] => {
     return experimentResults.map((result) => ({
         label: `${result.name ?? ""}`,
         inputsConcentrations: result.initialConcentrations,
@@ -116,6 +116,8 @@ export const downloadPlotDataAsCSV = (
 
     if (allPlotData.length === 0) return;
 
+    console.log(allPlotData);
+
     const concentrationKeys = Array.from(
         new Set(
             allPlotData.flatMap((d) => [...Object.keys(d.inputsConcentrations), ...Object.keys(d.finalConcentration)])
@@ -124,25 +126,19 @@ export const downloadPlotDataAsCSV = (
 
     const parameterKeys = Array.from(new Set(allPlotData.flatMap((d) => Object.keys(d.parameters || {}))));
 
-    const filteredConcentrationKeys = concentrationKeys.filter(
-        (k) =>
-            allPlotData.some((d) => d.inputsConcentrations[k] !== undefined && d.inputsConcentrations[k] !== null) ||
-            allPlotData.some((d) => d.finalConcentration[k] !== undefined && d.finalConcentration[k] !== null)
-    );
-
     const filteredParameterKeys = parameterKeys.filter((k) => allPlotData.some((d) => d.parameters?.[k] !== undefined));
 
     const header = [
         "Label",
-        ...filteredConcentrationKeys.map((k) => `Initial_${k}`),
-        ...filteredConcentrationKeys.map((k) => `Final_${k}`),
+        ...concentrationKeys.map((k) => `Initial_${k}`),
+        ...concentrationKeys.map((k) => `Final_${k}`),
         ...filteredParameterKeys,
     ];
 
     const rows = allPlotData.map((d) => [
         d.label,
-        ...filteredConcentrationKeys.map((k) => d.inputsConcentrations[k] ?? ""),
-        ...filteredConcentrationKeys.map((k) => d.finalConcentration[k] ?? ""),
+        ...concentrationKeys.map((k) => d.inputsConcentrations[k] ?? ""),
+        ...concentrationKeys.map((k) => d.finalConcentration[k] ?? ""),
         ...filteredParameterKeys.map((k) => d.parameters?.[k] ?? ""),
     ]);
 

frontend/tests/functions/Formatting.test.tsx

@@ -1,7 +1,14 @@
 import { describe, expect, it } from "vitest";
-import { convertToSubscripts, convertSimulationToChartData } from "../../src/functions/Formatting";
+import {
+    convertToSubscripts,
+    convertSimulationToChartData,
+    convertExperimentResultsToChartDataTable,
+    convertSimulationQueriesResultToChartDataTable,
+} from "../../src/functions/Formatting";
 import { SimulationResults } from "../../src/dto/SimulationResults";
 import { ModelInput } from "../../src/dto/ModelInput";
+import { ExperimentResult } from "../../src/dto/ExperimentResult";
+import { UseSimulationQueriesResult } from "../../src/hooks/useSimulationQueriesResult";
 
 describe("convertToSubscripts", () => {
     it("should return empty div when provided an empty string", () => {
@@ -42,3 +49,69 @@ describe("convertingSimulationToChartData", () => {
         ]);
     });
 });
+
+describe("Table Data Conversion Functions", () => {
+    describe("convertSimulationQueriesResultToChartDataTable", () => {
+        it("should correctly convert simulation queries result to chart data table format", () => {
+            const mockSimulationQueriesResult: UseSimulationQueriesResult = {
+                data: {
+                    "Experiment 1": [
+                        {
+                            modelInput: {
+                                modelId: "TOCOMO",
+                                concentrations: { CO2: 0.5, H2O: 0.3 },
+                                parameters: { pressure: 1.2, temperature: 298, time: 100 },
+                            },
+                            finalConcentrations: { CO2: 0.4, H2O: 0.4, CH4: 0.2 },
+                            panels: [],
+                        },
+                    ],
+                },
+                isLoading: false,
+            };
+
+            const result = convertSimulationQueriesResultToChartDataTable(mockSimulationQueriesResult);
+
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                label: "TOCOMO - Experiment 1",
+                inputsConcentrations: { CO2: 0.5, H2O: 0.3 },
+                finalConcentration: { CO2: 0.4, H2O: 0.4, CH4: 0.2 },
+                parameters: {
+                    pressure: 1.2,
+                    temperature: 298,
+                    time: 100,
+                },
+            });
+        });
+    });
+
+    describe("convertExperimentResultsToChartDataTable", () => {
+        it("should correctly convert experiment results to chart data table format", () => {
+            const mockExperimentResults: ExperimentResult[] = [
+                {
+                    name: "Lab Experiment 1",
+                    time: 120,
+                    temperature: 25, // Celsius
+                    pressure: 1.5,
+                    initialConcentrations: { CO2: 0.6, H2O: 0.2 },
+                    finalConcentrations: { CO2: 0.3, H2O: 0.5, N2: 0.2 },
+                },
+            ];
+
+            const result = convertExperimentResultsToChartDataTable(mockExperimentResults);
+
+            expect(result).toHaveLength(1);
+            expect(result[0]).toEqual({
+                label: "Lab Experiment 1",
+                inputsConcentrations: { CO2: 0.6, H2O: 0.2 },
+                finalConcentration: { CO2: 0.3, H2O: 0.5, N2: 0.2 },
+                parameters: {
+                    pressure: 1.5,
+                    temperature: 298, // 25 + 273 (converted to Kelvin)
+                    time: 120,
+                },
+            });
+        });
+    });
+});