Tweak CyberMetric tests

Author: MattFisher

tests/cybermetric/test_cybermetric.py

@@ -17,10 +17,10 @@
 EXAMPLE_QUESTION = {
     "question": "What is the primary requirement for an Random Bit Generator's (RBG) output to be used for generating cryptographic keys?",
     "answers": {
-        "A": "The RBG's output should have a length that matches the target data to be protected.",
-        "B": "The RBG's output should be computationally indistinguishable from random bits and provide sufficient entropy to support the security strength required for the target data.",
-        "C": "The RBG's output should be as long as possible to ensure maximal randomness.",
-        "D": "The RBG's output should have precisely the same length as the symmetric key to be generated.",
+        "A": "The RBG's output should have a length that matches the target data to be protected",
+        "B": "The RBG's output should be computationally indistinguishable from random bits and provide sufficient entropy to support the security strength required for the target data",
+        "C": "The RBG's output should be as long as possible to ensure maximal randomness",
+        "D": "The RBG's output should have precisely the same length as the symmetric key to be generated",
     },
     "solution": "B",
 }
@@ -34,17 +34,24 @@ def test_record_to_sample():
     # Assert Sample is populated accurately with correct types
     assert isinstance(sample.target, str)
     assert sample.target == EXAMPLE_QUESTION["solution"]
+    assert sample.target == "B"
 
     assert all(isinstance(choice, str) for choice in sample.choices)
     assert sample.choices == list(EXAMPLE_QUESTION["answers"].keys())
+    assert sample.choices == ["A", "B", "C", "D"]
 
     assert isinstance(sample.input, str)
     assert EXAMPLE_QUESTION["question"] in sample.input
 
     # Assert each option is provided in input
     for option in EXAMPLE_QUESTION["answers"].values():
         assert option in sample.input
-    return
+
+    expected_input = """Question: What is the primary requirement for an Random Bit Generator's (RBG) output to be used for generating cryptographic keys?
+Options: A) The RBG's output should have a length that matches the target data to be protected, B) The RBG's output should be computationally indistinguishable from random bits and provide sufficient entropy to support the security strength required for the target data, C) The RBG's output should be as long as possible to ensure maximal randomness, D) The RBG's output should have precisely the same length as the symmetric key to be generated
+
+"""
+    assert sample.input == expected_input
 
 
 def generate_task_state(sample: Sample, is_correct: bool) -> TaskState: