class OutputCollector:
    def __init__(self):
        from unit_tests.myStringIO import StringIO
        self.stream = StringIO()
        self.getvalue = self.stream.getvalue

    def write(self, data):
        self.stream.write(data)

    def should_be(self, expected):
        assert_equals_with_unidiff(expected, self.output())

    def output(self):
        return self.stream.getvalue()
class OutputCollector:
    def __init__(self):
        from unit_tests.myStringIO import StringIO
        self.stream = StringIO()
        self.getvalue = self.stream.getvalue
    def write(self,data):
        self.stream.write(data)
    def assert_equal_to(self, expected):
        return self.should_be(expected)
    def should_be(self, expected):
        assert_equals_with_unidiff(expected, self.output())
    def should_match(self, regex):
        text = self.output()
        from nose.tools import assert_regexp_matches
        assert_regexp_matches(text, regex)
    def output(self):
        return self.stream.getvalue()
Beispiel #3
0
class OutputCollector:
    def __init__(self):
        from unit_tests.myStringIO import StringIO
        self.stream = StringIO()
        self.getvalue = self.stream.getvalue

    def write(self, data):
        self.stream.write(data)

    def assert_equal_to(self, expected):
        return self.should_be(expected)

    def should_be(self, expected):
        assert_equals_with_unidiff(expected, self.output())

    def should_match(self, regex):
        text = self.output()
        from nose.tools import assert_regexp_matches
        assert_regexp_matches(text, regex)

    def output(self):
        return self.stream.getvalue()