import unittest from context import measprocess as mpc import os import pickle def extract_scan(path): filepath = os.path.join(path, 'ofdmscan.p0') with open(filepath, "rb") as fh: data = pickle.load(fh) return data[0][7][1]['rec'][0], data[0][7][1]['ot']['Band'], data[0][7]['ot']['Ch'] def extract_gps(path): filepath = os.path.join(path, 'gps.p0') with open(filepath, "rb") as fh: data = pickle.load(fh) return data[0]['ot'] class TestSum(unittest.TestCase): def test_synchronize(self): path = [ f"tests/example_files/", 'scanner' ] meas, band, channel = extract_scan( os.path.join(*path) ) meas['band'] = band meas['channel'] = channel gps = extract_gps( os.path.join(*path) ) merged = mpc.preprocess.synchronize_df(meas, gps, 'Datetime', 'fk_gps', ['Lon.', 'Lat.', 'Quality']) print(merged) #self.assertEqual(sum([1, 2, 3]), 6, "Should be 6") #def test_sum_tuple(self): # self.assertEqual(sum((1, 2, 2)), 6, "Should be 6") if __name__ == '__main__': unittest.main()