def test__mean_of_random_sample_with_no_reps(self): diffs = [1, 2, 3, 4, 5, 6, 7, 8, 9] timings = pd.DataFrame(data=self.neq_data_overlap) mock_read_csv = mock.Mock(spec=pd.read_csv) mock_read_csv.return_value = timings with mock.patch("tlsfuzzer.analysis.pd.read_csv", mock_read_csv): with mock.patch("tlsfuzzer.analysis._diffs", diffs): analysis = Analysis("/tmp") vals = analysis._mean_of_random_sample(0) self.assertEqual(len(vals), 0) self.assertEqual(vals, [])
def test__mean_of_random_sample(self): diffs = [1, 2, 3, 4, 5, 6, 7, 8, 9] timings = pd.DataFrame(data=self.neq_data_overlap) mock_read_csv = mock.Mock() mock_read_csv.return_value = timings with mock.patch("tlsfuzzer.analysis.Analysis.load_data", mock_read_csv): with mock.patch("tlsfuzzer.analysis._diffs", diffs): analysis = Analysis("/tmp") vals = analysis._mean_of_random_sample(10) self.assertEqual(len(vals), 10) avg = sum(vals) / len(vals) self.assertLessEqual(avg, 8) self.assertLessEqual(2, avg)