tested and tests for imgavg function

drcandacemakedamoore · May 16, 2021 · 853dece · 853dece
1 parent 5284dd6
commit 853dece
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/cleanX.py b/cleanX.py
@@ -1358,8 +1358,8 @@ def avg_image_maker(set_of_images):
 
     for example in set_of_images:
         example = cv2.imread(example, cv2.IMREAD_GRAYSCALE)
-        ht = image.shape[0]
-        wt = image.shape[1]
+        ht = example.shape[0]
+        wt = example.shape[1]
         list_h.append(ht)
         list_w.append(wt)
 

diff --git a/test/test_cleanx.py b/test/test_cleanx.py
@@ -197,4 +197,12 @@ def test_def_make_contour_image():
     picy1 = vovo
     defMkcont = cleanX.make_contour_image(picy1)
     assert len(defMkcont) > 0
-
+
+
+def test_avg_image_maker():
+    #set_of_images = glob.glob(*.jpg)
+    test_dfE = (os.path.join(image_directory,'test_sample_df.csv'))
+    test_df = pd.read_csv(test_dfE)
+    set_of_images = image_directory + '/' + test_df.image_path.dropna()
+    tab = cleanX.avg_image_maker(set_of_images)  
+    assert tab.shape[0] > 2