Fix a bug with deleting the first element in SparseVector that left an extra

empty partition.
Add extra checking to Partitioning and turn on checking for UnitTester.

1 files changed, 32 insertions, 0 deletions

diff --git a/src/Partitioning.h b/src/Partitioning.h
index c6951388e..608a3a58c 100644
--- a/src/Partitioning.h
+++ b/src/Partitioning.h
@@ -110,6 +110,10 @@ public:
 		return static_cast<T>(body->Length())-1;
 	}
 
+	T Length() const noexcept {
+		return PositionFromPartition(Partitions());
+	}
+
 	void InsertPartition(T partition, T pos) {
 		if (stepPartition < partition) {
 			ApplyStep(partition);
@@ -196,6 +200,34 @@ public:
 	void DeleteAll() {
 		Allocate(body->GetGrowSize());
 	}
+
+	void Check() const {
+#ifdef CHECK_CORRECTNESS
+		if (Length() < 0) {
+			throw std::runtime_error("Partitioning: Length can not be negative.");
+		}
+		if (Partitions() < 1) {
+			throw std::runtime_error("Partitioning: Must always have 1 or more partitions.");
+		}
+		if (Length() == 0) {
+			if ((PositionFromPartition(0) != 0) || (PositionFromPartition(1) != 0)) {
+				throw std::runtime_error("Partitioning: Invalid empty partitioning.");
+			}
+		} else {
+			// Positions should be a strictly ascending sequence
+			for (T i = 0; i < Partitions(); i++) {
+				const T pos = PositionFromPartition(i);
+				const T posNext = PositionFromPartition(i+1);
+				if (pos > posNext) {
+					throw std::runtime_error("Partitioning: Negative partition.");
+				} else if (pos == posNext) {
+					throw std::runtime_error("Partitioning: Empty partition.");
+				}
+			}
+		}
+#endif
+	}
+
 };