def test_calc_data_point_locations_custom_spacing(self): """_calc_data_point_locations() should return an array containing the x-axis locations for each data point, spaced according to a custom spacing scheme.""" locs = _calc_data_point_locations([3, 4, 10, 12], 4, 2, 0.25, 0.75) self.assertEqual(locs, array([3.75, 5.0, 12.5, 15.0]))
def test_calc_data_point_locations_default_spacing(self): """_calc_data_point_locations() should return an array containing the x-axis locations for each data point, evenly spaced from 1..n.""" locs = _calc_data_point_locations(None, 4, 2, 0.25, 0.5) self.assertEqual(locs, array([1.0, 2.0, 3.0, 4.0]))