def test_remove_first_histogram(self, rotate):
"""
Tests that removing the first histogram works.
"""
item = HistogramItem(rotate=rotate)
data = np.random.random(100)
item.add_histogram(data)
data2 = np.random.random(200)
item.add_histogram(data2)
data_item2 = item.listDataItems()[1]
item.remove_item_at(0)
self._assert_histogram_item_array_lengths(item, 1)
data_item_left = item.listDataItems()[0]
self.assertEqual(data_item2, data_item_left)
def test_add_histogram(self, rotate):
"""
Tests adding a histogram.
"""
item = HistogramItem(rotate=rotate)
# Assert no data items.
self._assert_histogram_item_array_lengths(item, 0)
data = np.random.random(100)
item.add_histogram(data)
data_items = item.listDataItems()
self._assert_histogram_item_array_lengths(item, 1)
