python handler for battle.hexlet.io

checker.py

import atexit
import fileinput
import os
import json
import unittest

class TestSolution(unittest.TestCase):
    maxDiff = None

    def test_solution(self):
        checks = [json.loads(line) for line in fileinput.input()]

        import solution

        for check in checks:
            expected = check['expected']
            arguments = check['arguments']
            if 'check' in check:
                atexit.register(onexit, check['check'])
            else:
                msg = "Arguments was: %s" % arguments
                self.assertEqual(expected, solution.solution(*arguments), msg)

def onexit(msg):
    print(msg)

if __name__ == '__main__':
    unittest.main()

data.json

{"expected": {"one": 5, "two": 5}, "arguments": [["one", "two"], 5]}
{"expected": {"one": 3}, "arguments": [["one"], 3]}

Dockerfile

FROM hexlet/hexlet-base

RUN apt-install python-pip python-dev
RUN apt-install python3-pip python3-dev
RUN pip3 install pytest

ENV PYTHONDONTWRITEBYTECODE 1

Makefile

test:
	@cat data.jsons | sudo -u nobody python3 checker.py

.PHONY: test

solution.py

# BEGIN python3
def solution(l, d):
    return {v: d for v in l}
# END