diff --git a/src/test/java/Team5/SmartTowns/CalculatorTest.java b/src/test/java/Team5/SmartTowns/CalculatorTest.java
index 1aea1c6ac81241ecd401b728217b416d615e2d1b..33b9fdac1f216402b180ebbd9e346806ca11d201 100644
--- a/src/test/java/Team5/SmartTowns/CalculatorTest.java
+++ b/src/test/java/Team5/SmartTowns/CalculatorTest.java
@@ -9,7 +9,7 @@ import org.springframework.core.NestedRuntimeException;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertThrows;
-import org.junit.platform.engine.TestDescriptor;
+//import org.junit.platform.engine.TestDescriptor;
 //import org.junit.platform.commons.util.Preconditions;
 import org.apache.commons.logging.LogFactory;
 @SpringBootTest
diff --git a/src/test/java/Team5/SmartTowns/DBUnitTestCase.java b/src/test/java/Team5/SmartTowns/DBUnitTestCase.java
index 7d64da18812b63cbd5fdd2d223de969fced81794..646b90a8cd6d02cf87f01d97fc7100023abe7ebb 100644
--- a/src/test/java/Team5/SmartTowns/DBUnitTestCase.java
+++ b/src/test/java/Team5/SmartTowns/DBUnitTestCase.java
@@ -1,6 +1,9 @@
 package Team5.SmartTowns;
 
 import org.junit.jupiter.api.Test;
+
+//import static org.testing.AssertJUnit.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 //import org.springframework.boot.test.context.SpringBootTest;
 //import org.testng.annotations.Test;
 //import org.testng.annotations.Test;
@@ -8,7 +11,7 @@ import org.junit.jupiter.api.Test;
 //import org.junit.platform.engine.TestDescriptor;
 
 //import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.testng.AssertJUnit.assertEquals;
+
 
 public class DBUnitTestCase {
     int bb=4;
diff --git a/src/test/java/Team5/SmartTowns/Test5.java b/src/test/java/Team5/SmartTowns/Test5.java
index 4ea4af12de0eb1e1dbe9f5dff8b9d2778d711d50..75877470dd7f8801c67edd38fa68b4edf56fafaf 100644
--- a/src/test/java/Team5/SmartTowns/Test5.java
+++ b/src/test/java/Team5/SmartTowns/Test5.java
@@ -1,39 +1,39 @@
-//package Team5.SmartTowns;
-//
-//import Team5.SmartTowns.Data.locationRepository;
-//import Team5.SmartTowns.Data.locationRepositoryJDBC;
-////import org.junit.jupiter.api.Test;
-//import org.springframework.beans.factory.annotation.Autowired;
-//import org.springframework.boot.test.autoconfigure.jdbc.JdbcTest;
-//import org.springframework.boot.test.context.SpringBootTest;
-//import org.springframework.jdbc.core.JdbcTemplate;
-//import org.springframework.test.context.jdbc.Sql;
-//import org.testng.annotations.Test;
-//
-////import static org.junit.jupiter.api.Assertions.assertEquals;
-//import static org.testng.Assert.assertEquals;
-//
-//@SpringBootTest
-////@JdbcTest
-////
-//@Sql({"schema.sql", "test-data.sql"})
-//public class Test5 {
-//
-//    @Autowired
-//    private locationRepository locationRepo;
-//
-//
-//    @Test
-////	test 1
-//    void whenInjectInMemoryDataSource_thenReturnCorrectEmployeeCount() {
-////        locationRepositoryJDBC employeeDAO = new locationRepositoryJDBC(ajdbc);
-//        int aa=2;
-//        int bb=2;
-////        employeeDAO.setJdbc(ajdbc);
-//        assertEquals(4, (aa+bb));
-//
-//    }
-//
-//
-//
-//}
+package Team5.SmartTowns;
+
+import Team5.SmartTowns.Data.locationRepository;
+import Team5.SmartTowns.Data.locationRepositoryJDBC;
+//import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.jdbc.JdbcTest;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.test.context.jdbc.Sql;
+import org.testng.annotations.Test;
+
+//import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.testng.Assert.assertEquals;
+
+@SpringBootTest
+//@JdbcTest
+//
+@Sql({"schema.sql", "test-data.sql"})
+public class Test5 {
+
+    @Autowired
+    private locationRepository locationRepo;
+
+
+    @Test
+//	test 1
+    void whenInjectInMemoryDataSource_thenReturnCorrectEmployeeCount() {
+//        locationRepositoryJDBC employeeDAO = new locationRepositoryJDBC(ajdbc);
+        int aa=2;
+        int bb=2;
+//        employeeDAO.setJdbc(ajdbc);
+        assertEquals(4, (aa+bb));
+
+    }
+
+
+
+}