Update test/model.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Dec 12, 2024 · 798014d · 798014d
1 parent 500ddfe
commit 798014d
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/model.jl b/test/model.jl
@@ -13,8 +13,14 @@
         @test model.transformed == deserialized.transformed
         @test model.untransformed_param_length == deserialized.untransformed_param_length
         @test model.transformed_param_length == deserialized.transformed_param_length
-        @test all(model.untransformed_var_lengths[k] == deserialized.untransformed_var_lengths[k] for k in keys(model.untransformed_var_lengths))
-        @test all(model.transformed_var_lengths[k] == deserialized.transformed_var_lengths[k] for k in keys(model.transformed_var_lengths))
+        @test all(
+            model.untransformed_var_lengths[k] == deserialized.untransformed_var_lengths[k]
+            for k in keys(model.untransformed_var_lengths)
+        )
+        @test all(
+            model.transformed_var_lengths[k] == deserialized.transformed_var_lengths[k] for
+            k in keys(model.transformed_var_lengths)
+        )
         @test Set(model.parameters) == Set(deserialized.parameters)
         # skip testing g
         @test model.model_def === deserialized.model_def