def test_create_version(self): version = versions[0] base_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) vars_filename = os.path.join(base_dir, version[1], version[2]) compile_filename = os.path.join(base_dir, version[1], version[3]) with capture_stdout(): call_command('createversion', version[0], vars_filename, compile_filename) self.assertEqual(1, Version.objects.count())
def test_pprint(self): d = {"foo": "bar", "thing": 3} expected = """{\n "foo": "bar",\n "thing": 3\n}\n""" from wrench.contexts import capture_stdout with capture_stdout() as output: pprint(d) self.assertEqual(expected, output.getvalue())
def test_pprint(self): d = { 'foo': 'bar', 'thing': 3, } expected = """{\n "foo": "bar",\n "thing": 3\n}\n""" from wrench.contexts import capture_stdout with capture_stdout() as output: pprint(d) self.assertEqual(expected, output.getvalue())
def test_module_main(self): # write VARS_FILE into a file that gets cleaned up with capture_stdout(), temp_file() as filename: f = open(filename, 'w') f.write(VARS_FILE) f.close() import sys keep = sys.argv try: sys.argv[1] = filename except IndexError: sys.argv.append(filename) runpy.run_module('bseditor.conv', run_name='__main__') sys.argv = keep
def test_capture_stdout(self): with capture_stdout() as capture: sys.stdout.write('foo\n') self.assertEqual(capture.getvalue(), 'foo\n')
def test_default_version(self): with capture_stdout(): call_command('defaultversion') self.assertEqual(1, Version.objects.count())