Added unittest on tables

1 files changed, 26 insertions, 0 deletions

diff --git a/src/lib.rs b/src/lib.rs
index e1db5e5..7ef9b76 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -314,3 +314,29 @@ macro_rules! ptable {
 		}

 	);

 }

+

+#[cfg(test)]

+mod tests {

+	

+	use Table;

+	use row::Row;

+	use cell::Cell;

+	

+	#[test]

+	fn table() {

+		let mut table = Table::new();

+		table.add_row(Row::new(vec![Cell::new("a"), Cell::new("bc"), Cell::new("def")]));

+		table.add_row(Row::new(vec![Cell::new("def"), Cell::new("bc"), Cell::new("a")]));

+		table.set_titles(Row::new(vec![Cell::new("t1"), Cell::new("t2"), Cell::new("t3")]));

+		let out = "\

++-----+----+-----+

+| t1  | t2 | t3  |

++=====+====+=====+

+| a   | bc | def |

++-----+----+-----+

+| def | bc | a   |

++-----+----+-----+

+";

+		assert_eq!(table.to_string().replace("\r\n", "\n"), out);

+	}

+}