def test_remove_rows_from_the_middle(self):
model = TimePatternModel(TimePattern(['a', 'b', 'c'],
[-5.0, 3.0, 7.0]))
self.assertTrue(model.removeRows(1, 1))
self.assertEqual(len(model.value), 2)
self.assertEqual(model.value.indexes, ['a', 'c'])
numpy.testing.assert_equal(model.value.values, np.array([-5.0, 7.0]))
def test_cannot_remove_all_rows(self):
model = TimePatternModel(TimePattern(['a', 'b'], [-5.0, 7.0]))
self.assertTrue(model.removeRows(0, 2))
self.assertEqual(len(model.value), 1)
self.assertEqual(model.value.indexes, ['a'])
numpy.testing.assert_equal(model.value.values, [-5.0])
def test_remove_rows_from_the_beginning(self):
model = TimePatternModel(TimePattern(['a', 'b'], [-5.0, 7.0]))
self.assertTrue(model.removeRows(0, 1))
self.assertEqual(len(model.value), 1)
self.assertEqual(model.value.indexes, ['b'])
numpy.testing.assert_equal(model.value.values, np.array([7.0]))
def test_removing_last_row_fails(self):
model = TimePatternModel(TimePattern(['a'], [-5.0]))
self.assertFalse(model.removeRows(0, 1))