Test main done (initial version).

Author: MKaptein

test/test_main.py

@@ -46,20 +46,29 @@ def test_display():
 
 def test_scblbl_input():
     example = {"in": np.array([1,2,3,4]).astype(np.int32)}
-    result = sclbl_input(example)
+    result = sclbl_input(example, _verbose=False)
     assert result == '{"input": CAQQBkoQAQAAAAIAAAADAAAABAAAAA==, "type":"pb"}', "PB output not correct."
 
     example = {"x1": np.array([1,2,3,4]).astype(np.int32), "x2": np.array([1,2,3,4]).astype(np.int32)}
-    result = sclbl_input(example)
-    assert result == '{"input": ["CAQQBkoQAQAAAAIAAAADAAAABAAAAA==","CAQQBkoQAQAAAAIAAAADAAAABAAAAA=="], "type":"pb"}', "PB output 2 not correct."
+    result = sclbl_input(example, _verbose=False)
+    assert result == '{"input": ["CAQQBkoQAQAAAAIAAAADAAAABAAAAA==","CAQQBkoQAQAAAAIAAAADAAAABAAAAA=="], "type":"pb"}',\
+        "PB output 2 not correct. "
 
-    # todo: Do raw tests.
+    example = {"in": np.array([1,2,3,4]).astype(np.int32)}
+    result = sclbl_input(example, "raw", _verbose=False)
+    assert result == '{"input": AQAAAAIAAAADAAAABAAAAA==, "type":"raw"}', "Raw output not correct."
+
+    example = {"x1": np.array([1,2,3,4]).astype(np.int32), "x2": np.array([1,2,3,4]).astype(np.int32)}
+    result = sclbl_input(example, "raw", _verbose=False)
+    assert result == '{"input": ["AQAAAAIAAAADAAAABAAAAA==","AQAAAAIAAAADAAAABAAAAA=="], "type":"raw"}',\
+        "Raw output 2 not correct. "
 
 
 def test_list_data_types():
     test = list_data_types()
     assert test, "Data types should be listed."
 
+
 def test_list_operators():
     test = list_operators()
     assert test, "Operators should be listed."