operation: io: Run Doctest Examples

Fixes: #524

Author: 0dust

CHANGELOG.md

@@ -6,7 +6,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 ### Added
-- Operations for taking input from the user(AcceptUserInput) and for printing the output(print_output)
+- Operations for taking input from the user `AcceptUserInput` and for printing the output `print_output`
 - Hugging Face Transformers tensorflow based NER models.
 - PNG ConfigLoader for reading images as arrays to predict using MNIST trained models
 - Docstrings and doctestable examples to `record.py`.

dffml/operation/io.py

@@ -52,6 +52,7 @@ class AcceptUserInput(OperationImplementation):
     ++++++++
 
     The following example shows how to use AcceptUserInput.
+
     >>> dataflow = DataFlow.auto(AcceptUserInput, GetSingle)
     >>> dataflow.seed.append(
     ...     Input(
@@ -106,6 +107,7 @@ async def print_output(data: str):
     ++++++++
 
     The following example shows how to use print_output.
+
     >>> dataflow = DataFlow.auto(print_output, GetSingle)
     >>> inputs = [
     ...     Input(
@@ -121,4 +123,4 @@ async def print_output(data: str):
     print_output example
     String to be printed is 'print_output example'
     """
-    print("\n" + data)
+    print(data)

docs/doctest_header.py

@@ -7,7 +7,9 @@
 import inspect
 import asyncio
 import tempfile
+import builtins
 import functools
+from unittest import mock
 
 # Create a temporary directory for test to run in
 DOCTEST_TEMPDIR = tempfile.mkdtemp()
@@ -16,14 +18,18 @@
 # Change the current working directory to the temporary directory
 os.chdir(DOCTEST_TEMPDIR)
 
-from dffml_model_scikit import *
 from dffml import *
 from dffml.base import *
 from dffml.record import *
 from dffml.df.base import *
 from dffml.df.types import *
-from dffml.df.memory import *
 from dffml.util.net import *
+from dffml.df.memory import *
+from dffml_model_scikit import *
+from dffml.operation.io import *
+from dffml.source.memory import *
 from dffml.operation.output import *
 from dffml.operation.dataflow import *
-from dffml.source.memory import *
+
+# Used for mocking input() for AcceptUserInput operation.
+mock.patch("builtins.input", return_value="Data flow is awesome").start()