class ChartTest(ComparisonTestCase): """ Test for the Chart baseclass methods. """ def setUp(self): self.xs = range(11) self.ys = np.linspace(0, 1, 11) self.chart = Chart(zip(self.xs, self.ys)) self.curve = Curve(zip(self.xs, self.ys)) def test_yvalue_constructor(self): ys = np.linspace(0, 1, 11) Chart(ys) def test_chart_index(self): self.assertEqual(self.chart[5], self.ys[5]) def test_chart_slice(self): chart_slice = Curve(zip(range(5, 9), np.linspace(0.5,0.8, 4))) self.assertEqual(self.curve[5:9], chart_slice) def test_chart_closest(self): closest = self.chart.closest([0.51, 1, 9.9]) self.assertEqual(closest, [1., 1., 10.]) def test_chart_reduce(self): mean = self.chart.reduce(x=np.mean) itable = ItemTable(OrderedDict([('y', np.mean(self.ys))])) self.assertEqual(mean, itable) def test_chart_sample(self): samples = self.chart.sample([0, 5, 10]).values() self.assertEqual(samples, [(0,), (0.5,), (1,)])
class ChartTest(ComparisonTestCase): """ Test for the Chart baseclass methods. """ def setUp(self): self.xs = range(11) self.ys = np.linspace(0, 1, 11) self.chart = Chart(zip(self.xs, self.ys)) self.curve = Curve(zip(self.xs, self.ys)) def test_yvalue_constructor(self): ys = np.linspace(0, 1, 11) Chart(ys) def test_chart_index(self): self.assertEqual(self.chart[5], self.ys[5]) def test_chart_slice(self): chart_slice = Curve(zip(range(5, 9), np.linspace(0.5, 0.8, 4))) self.assertEqual(self.curve[5:9], chart_slice) def test_chart_closest(self): closest = self.chart.closest([0.51, 1, 9.9]) self.assertEqual(closest, [1., 1., 10.]) def test_chart_reduce(self): mean = self.chart.reduce(x=np.mean) itable = ItemTable(OrderedDict([('y', np.mean(self.ys))])) self.assertEqual(mean, itable) def test_chart_sample(self): samples = self.chart.sample([0, 5, 10]).values() self.assertEqual(samples, [(0, ), (0.5, ), (1, )])
def setUp(self): self.xs = range(11) self.ys = np.linspace(0, 1, 11) self.chart = Chart(zip(self.xs, self.ys)) self.curve = Curve(zip(self.xs, self.ys))
def test_yvalue_constructor(self): ys = np.linspace(0, 1, 11) Chart(ys)