Update image versions and tests

Author: vinkj1

tests/test_pytorch_env.py

@@ -87,7 +87,7 @@ def test_dockerfile(self):
             lines = list(map(lambda x: x.strip(), fin.readlines()))
         dockerfile_text = Path(dockerfile_path).read_text()
 
-        assert lines[1] == "FROM registry.git.vgregion.se/aiplattform/images/pytorch:0.3.1"
+        assert lines[1] == "FROM registry.git.vgregion.se/aiplattform/images/pytorch:0.3.2"
         assert lines[-1] == 'WORKDIR /workspace'
         assert "ADD http://aiav2.vgregion.se/VGC%20Root%20CA%20v2.crt /tmp/vgc_root.der" in lines
         assert "ADD http://aiav2.vgregion.se/VGC%20Issuing%201%20CA%20v2.crt /tmp/vgc_issuing1.der" in lines

tests/test_tensorflow_env.py

@@ -87,7 +87,7 @@ def test_dockerfile(self):
             lines = list(map(lambda x: x.strip(), fin.readlines()))
         dockerfile_text = Path(dockerfile_path).read_text()
 
-        assert lines[0] == "FROM registry.git.vgregion.se/aiplattform/images/tensorflow:0.2.0"
+        assert lines[0] == "FROM registry.git.vgregion.se/aiplattform/images/tensorflow:0.2.1"
         assert lines[-1] == 'WORKDIR /workspace'
         assert "ADD http://aiav2.vgregion.se/VGC%20Root%20CA%20v2.crt /tmp/vgc_root.der" in lines
         assert "ADD http://aiav2.vgregion.se/VGC%20Issuing%201%20CA%20v2.crt /tmp/vgc_issuing1.der" in lines