1 files changed, 8 insertions, 4 deletions
diff --git a/src/template.rs b/src/template.rs
index e50ad21..2f5da4e 100644
--- a/src/template.rs
+++ b/src/template.rs
@@ -254,13 +254,13 @@ mod function {
 mod table {
     use super::*;
 
-    #[derive(Debug, Serialize, Deserialize, Clone)]
+    #[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
     pub struct TableRow {
         pub index: i32,
         pub columns: Vec<Json>,
     }
 
-    #[derive(Debug, Serialize, Deserialize, Clone)]
+    #[derive(Debug, Serialize, Deserialize, Clone, PartialEq)]
     pub struct DisplayTable {
         pub num_rows: usize,
         pub num_columns: usize,
@@ -332,7 +332,11 @@ mod tests {
         ];
 
         let table = table::apply(&columns, &d).unwrap();
-        assert_eq!(table.headings, vec!("Name", "Distance"));
-        assert_eq!(table.rows, vec!(vec!(json!("Nick"), json!("2.0"))));
+        assert_eq!(table.headings.columns, vec!(json!("Name"), json!("Distance")));
+        assert_eq!(table.rows, vec!(
+            table::TableRow {
+                index: 2,
+                columns: vec!(json!("Nick"), json!("2.0"))
+            }));
     }
 }