def test_checkTimes(self): TEST_FILE = 'test_data/test_data.csv' testPointSet = pointSet.PointSet() testPointSet.loadData(TEST_FILE) # with bad timestamps testPointSet.removeTimeErrors() self.assertEqual(5, testPointSet.getLength()) TEST_TIME_FILE = 'test_data/test_timedata.csv' testTimeSet = pointSet.PointSet() testTimeSet.loadData(TEST_TIME_FILE) # without timestamps testTimeSet.removeTimeErrors() self.assertEqual(2, testTimeSet.getLength())
def main(): CSV_FILE = 'data/data_points.csv' ps = pointSet.PointSet() ps.loadData(CSV_FILE) ps.removeOutliers() ps.plotPoints() ps.printDetails()
def test_removeOutliers(self): TEST_FILE = 'test_data/test_data.csv' testPointSet = pointSet.PointSet() testPointSet.loadData(TEST_FILE) testPointSet.removeOutliers() testPointSet.printDetails() self.assertEqual(3, testPointSet.getLength())
def test_checkDist(self): TEST_FILE = 'test_data/test_data.csv' testPointSet = pointSet.PointSet() testPointSet.loadData(TEST_FILE) self.assertEqual(5, testPointSet.getLength()) testPointSet.setDistances() self.assertEqual(4, testPointSet.getLength())
def test_calcDist(self): TEST_FILE = 'test_data/test_data.csv' testPointSet = pointSet.PointSet() testPointSet.loadData(TEST_FILE) self.assertEqual(157.27478863265398, testPointSet.getDistance(testPointSet.pointSet[0], testPointSet.pointSet[1]))
def test_getMAD(self): TEST_FILE = 'test_data/test_data.csv' testPointSet = pointSet.PointSet() testPointSet.loadData(TEST_FILE) testPointSet.setDistances() self.assertEqual(0.04789525410973283, testPointSet.getMAD())