def __init__(self, pkg_dir):
self.pkg_dir = pkg_dir
self.data_acquisition = DataAcquisition()
self.data_processing = DataProcessing()
self.model_engineering = ModelEngineering(self.pkg_dir)
self.cam_streamer = CamStreamer()
#self.image_subscriber = ImageSubscriber()
self.acquire_data()
self.model_engineering.knn_fit(self.data_acquisition.trn_wh)
self.selected_face = None
self.pub_img = rospy.Publisher('/person_recognition/image/compressed',
CompressedImage,
queue_size=1)
self.pub_txt = rospy.Publisher('/person_recognition/crowd',
String,
queue_size=10)
def __init__(self, *args, **kwargs):
unittest.TestCase.__init__(self, *args, **kwargs)
self.pkg_dir = os.path.join('..')
self.data_acquisition = DataAcquisition()
def test_fetch_all_data(self):
data_acquisition = DataAcquisition()
df = data_acquisition.fetch_all_data()
self.assertGreater(df.shape[0], 0)
self.assertGreater(df.shape[1], 0)
def test_load_general_stats(self):
data_acquisition = DataAcquisition()
df = data_acquisition.load_general_stats()
self.assertGreater(df.shape[0], 0)
self.assertGreater(df.shape[1], 0)
def test_fetch_fetch_bundesland_rki(self):
data_acquisition = DataAcquisition()
df = data_acquisition.fetch_bundesland_rki("Bayern")
self.assertGreater(df.shape[0], 0)
self.assertGreater(df.shape[1], 0)
def test_fetch_germany_morgenpost(self):
data_acquisition = DataAcquisition()
df = data_acquisition.fetch_germany_morgenpost()
self.assertGreater(df.shape[0], 0)
self.assertGreater(df.shape[1], 0)