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))