def test_networkLoad_Time_withoutDebugMode(self):
global fmdlAlgo_Hydraulic
algoLoadStartTime = datetime.now()
fmdlAlgo_Hydraulic = FMDLAlgo(
'tests/networksAndImages-forUnitTesting/bbDetector.pb',
'tests/networksAndImages-forUnitTesting/roiDelineator.h5')
algoLoadTime = datetime.now() - algoLoadStartTime
self.assertIsNotNone(fmdlAlgo_Hydraulic)
self.assertLess(algoLoadTime, NETWORK_LOAD_TIME_THRESHOLD)
"roiBoundaryPointsReductionFactor": 0.01,
"minBoundingBoxAspectRatio": 1,
"maxBoundingBoxAspectRatio": 4,
"minObjectsRequired": [["bucket"]]
}
'''
##############################################################################
import PIL
import base64
from matplotlib import pyplot as plt
import io
import os
fmdlAlgo = FMDLAlgo(boxDetectorNetworkPath,
roiDelineatorNetworkPath,
debugMode=True)
if runOnSingleImage:
run_FMDLAlgo()
else:
TEST_IMAGE_PATHS = []
TEST_IMAGE_IDS = []
for fileName in os.listdir(testdirPath):
TEST_IMAGE_IDS.append(fileName)
TEST_IMAGE_PATHS.append(testdirPath + fileName)
results = {}
failedCounter = 0
successCounter = 0
def test_networkLoad_Exception_networkNotFound_roiDelineator(self):
with pytest.raises(Exception):
FMDLAlgo(
'tests/networksAndImages-forUnitTesting/bbDetector.pb',
'',
debugMode=True)
def test_networkLoad_Exception(self):
with pytest.raises(Exception):
FMDLAlgo(
'tests/networksAndImages-forUnitTesting/BadBoxDetectorNetwork.pb',
'tests/networksAndImages-forUnitTesting/BadRoiDelineatorNework.h5',
debugMode=True)