Fix unit test

dojoengine · Apr 15, 2024 · 0030dad · 0030dad
1 parent 856c5d1
commit 0030dad
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/bin/sozo/src/commands/build.rs b/bin/sozo/src/commands/build.rs
@@ -109,12 +109,12 @@ fn create_stats_table(contracts_statistics: Vec<ContractStatistics>) -> Table {
 
 #[cfg(test)]
 mod tests {
+    use super::{create_stats_table, BuildArgs};
     use dojo_test_utils::compiler::build_test_config;
+    use prettytable::format::consts::FORMAT_NO_LINESEP_WITH_TITLE;
     use prettytable::{format, Cell, Row, Table};
     use sozo_ops::statistics::ContractStatistics;
 
-    use super::{create_stats_table, BuildArgs};
-
     #[test]
     fn build_example_with_typescript_and_unity_bindings() {
         let config = build_test_config("../../examples/spawn-and-move/Scarb.toml").unwrap();
@@ -124,6 +124,7 @@ mod tests {
             typescript: true,
             unity: true,
             typescript_v2: true,
+            stats: true,
         };
         let result = build_args.run(&config);
         assert!(result.is_ok());
@@ -151,6 +152,7 @@ mod tests {
         ];
 
         let mut expected_table = Table::new();
+        expected_table.set_format(*FORMAT_NO_LINESEP_WITH_TITLE);
         expected_table.add_row(Row::new(vec![
             Cell::new_align("Contract", format::Alignment::CENTER),
             Cell::new_align("Bytecode size (felts)", format::Alignment::CENTER),