incorporate an AI model for prediction. Update unit tests
Change-Id: I0591691bb05ed0ef2262946d6668d8efd9a6d188
Signed-off-by: pceicicd <pekwatch746@gmail.com>
diff --git a/tests/test_lp.py b/tests/test_lp.py
index 054ef59..eafca5b 100644
--- a/tests/test_lp.py
+++ b/tests/test_lp.py
@@ -253,6 +253,8 @@
expected=sdl.get_uedata(self, "9876543")
assert expected==good_cell
+ expected=main.predict(self, expected)
+ assert expected=="Normal"
expected=sdl.get_uedata(self, "3456789")
try:
sdl.get_uedata(self, "1234567")