diff --git a/tests/v1.robot b/tests/v1.robot
new file mode 100644
index 0000000..86705b9
--- /dev/null
+++ b/tests/v1.robot
@@ -0,0 +1,32 @@
+*** Settings ***
+Library           SeleniumLibrary
+Library           Process
+
+*** Variables ***
+${URL}            http://localhost:8501
+${URL1}           http://localhost:8501/v1
+${BROWSER}        chrome
+
+*** Test Cases ***
+
+Test Head
+    Log To Console    test_head
+    Open Browser    ${URL}    ${BROWSER}
+    Sleep    5
+    Go To    ${URL1}
+    Sleep    5
+    Page Should Contain  AI Infrastructure for Model training
+
+    Page Should Contain Button  Train
+    Click Button    Train
+    Sleep           5
+    Click Button    Train
+#    Sleep           10
+#    Page Should Contain Button    Download trained model
+#    Click Button    Download trained model
+
+
+
+    Sleep    5
+
+    Close Browser
diff --git a/tests/v2.robot b/tests/v2.robot
new file mode 100644
index 0000000..c8364ec
--- /dev/null
+++ b/tests/v2.robot
@@ -0,0 +1,31 @@
+*** Settings ***
+Library           SeleniumLibrary
+Library           Process
+
+*** Variables ***
+${URL}            http://localhost:8501
+${URL2}           http://localhost:8501/v2
+${BROWSER}        chrome
+
+*** Test Cases ***
+
+Test Head
+    Log To Console    test_head
+    Open Browser    ${URL}    ${BROWSER}
+    Sleep    5
+    Go To    ${URL2}
+    Page Should Contain  AI Infrastructure for Model training
+
+    Page Should Contain Button  Train
+    Click Button    Train
+    Sleep           5
+    Click Button    Train
+    Sleep           10
+    Page Should Contain Button    Download trained model
+    Click Button    Download trained model
+
+
+
+    Sleep    5
+
+    Close Browser