fixing the scope command message test

AllenInstitute · May 7, 2024 · f712979 · f712979
1 parent 5e30bf0
commit f712979
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/test/test_scope_command.py b/test/test_scope_command.py
@@ -1,21 +1,21 @@
-from TEM_comms.msgs.scope import command
+from TEM_comms.msgs.scope import Command
 from pydantic import ValidationError
 import pytest
 
 def test_mag():
     with pytest.raises(ValidationError):
-        command(mag_mode="LM")
+        Command(mag_mode="LM")
 
     with pytest.raises(ValidationError):
-        command(mag=10)
+        Command(mag=10)
 
-    command()
-    command(mag_mode="LM", mag=1)
+    Command()
+    Command(mag_mode="LM", mag=1)
 
 def test_mag_mode():
     with pytest.raises(ValidationError):
-        command(mag_mode="test", mag=1)
+        Command(mag_mode="test", mag=1)
 
-    command(mag_mode="LM", mag=1)
-    command(mag_mode="MAG2", mag=1)
-    command(mag_mode="MAG2", mag=1)
+    Command(mag_mode="LM", mag=1)
+    Command(mag_mode="MAG2", mag=1)
+    Command(mag_mode="MAG2", mag=1)