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)