Allow assign to take a map

Co-authored-by: John Cotton <[email protected]>

Author: amclain

lib/scenic/scene.ex

@@ -391,11 +391,15 @@ defmodule Scenic.Scene do
   end
 
   @doc """
-  Convenience function to assign a list of values into a scene struct.
+  Convenience function to assign a list or map of values into a scene struct.
   """
-  @spec assign(scene :: Scene.t(), key_list :: Keyword.t()) :: Scene.t()
-  def assign(%Scene{} = scene, key_list) when is_list(key_list) do
-    Enum.reduce(key_list, scene, fn {k, v}, acc -> assign(acc, k, v) end)
+  @spec assign(scene :: Scene.t(), assigns :: Keyword.t() | map()) :: Scene.t()
+  def assign(%Scene{} = scene, assigns) when is_list(assigns) do
+    Enum.reduce(assigns, scene, fn {k, v}, acc -> assign(acc, k, v) end)
+  end
+
+  def assign(%Scene{} = scene, assigns) when is_map(assigns) do
+    %Scene{scene | assigns: Map.merge(scene.assigns, assigns)}
   end
 
   @doc """

test/scenic/scene_test.exs

@@ -195,6 +195,19 @@ defmodule Scenic.SceneTest do
     assert scene.assigns == %{one: 1, two: 2, three: "three"}
   end
 
+  test "assign assigns multiple values in a map", %{scene: scene} do
+    assert scene.assigns == %{}
+
+    scene =
+      Scene.assign(scene, %{
+        one: 1,
+        two: 2,
+        three: "three"
+      })
+
+    assert scene.assigns == %{one: 1, two: 2, three: "three"}
+  end
+
   test "get gets a value from assigns - works like map", %{scene: scene} do
     scene =
       Scene.assign(scene,