refactor more tests

Author: borisveis

.DS_Store

@@ -7,9 +7,7 @@
 
 def chat(prompt, maxtokens):
     return model.generate(prompt, max_tokens=100)
-
-class validate:
-    def cosine_similarity_score(sentence1, sentence2):
+def cosine_similarity_score(sentence1, sentence2):
         vectorizer = TfidfVectorizer()
         vectors = vectorizer.fit_transform([sentence1, sentence2])
         similarity = cosine_similarity(vectors[0], vectors[1])[0][0]

src/utils/framework.py

@@ -2,4 +2,4 @@
 
 
 def test_cos_sim():
-    assert framework.validate.cosine_similarity_score("abc","abc")==1
+    assert framework.cosine_similarity_score("abc","abc")==1

src/utils/similarity.py

@@ -5,18 +5,18 @@
 
 model = GPT4All(config.model)
 
-responsestates = framework.chat("how many states are in US", 100)
+response_states = framework.chat("how many states are in US", 100)
 expected_number_of_states_response = "There are 50 states"
 # cosinesimularity = framework.cosine_similarity(responsestates,expected_number_of_states_response)
 def test_cosinesimularity():
-        assert framework.validate.cosine_similarity_score(responsestates,expected_number_of_states_response) > 0.5
+        assert framework.cosine_similarity_score(response_states, expected_number_of_states_response) > 0.5
 
 
 # a positive test validating  reponse contains specific strings is included in the response(s)
 def test_contains():
 
-        assert (responsestates.__contains__("United States") and (responsestates.__contains__("50")
-                                                                  or responsestates.__contains__("fifty")))
+        assert (response_states.__contains__("United States") and (response_states.__contains__("50")
+                                                                   or response_states.__contains__("fifty")))
 
 #a positive test validating a specific string is not included in the respons(s)
 def test_notcontains():

src/utils/test_utils.py

@@ -8,8 +8,8 @@
 print("response:"+responsestates)
 numberofstates = "There are 50 states"
 print("cosine simularity score:")
-cosinesimularity = testFramwork.validate.cosine_similarity_score(responsestates, numberofstates)
-print(testFramwork.validate.cosine_similarity_score(responsestates, numberofstates))
+cosinesimularity = testFramwork.cosine_similarity_score(responsestates, numberofstates)
+print(testFramwork.cosine_similarity_score(responsestates, numberofstates))
 def test_cosinesimularity():
         print("cosinesimularity="+str(cosinesimularity))
         assert cosinesimularity>0.6