From 4e52addcaf5c3a1797a5d58b58586722d724ab83 Mon Sep 17 00:00:00 2001
From: kornelHub <67650984+kornelHub@users.noreply.github.com>
Date: Thu, 17 Jun 2021 04:06:26 +0200
Subject: [PATCH] small fix when math_char is '='

---
 engine/simulation.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/engine/simulation.py b/engine/simulation.py
index 03bf95f..3d6356e 100644
--- a/engine/simulation.py
+++ b/engine/simulation.py
@@ -106,6 +106,9 @@ def build_if_statement(first_indicator_short_name, first_indicator_options_list,
         second_indicator_period = second_indicator_period[1:-1]  # remove square bracket
         if_statement_second_part = f" data_df.iloc[x+{second_indicator_period}]['{second_indicator_collumn_name}']:"
 
+    if math_char == '=':
+        math_char += '='
+
     if_statement = if_statement_first_part + math_char + if_statement_second_part
     return if_statement