class PylsyTableTests(unittest.TestCase):
def setUp(self):
attributes=["name","age"]
self.table = PylsyTable(attributes)
def tearDown(self):
self.table = None
def testCreateTable(self):
name=["a", "b"]
self.table.add_data("name",name)
age=[1, 2]
self.table.add_data("age",age)
correct_file = open('correct.out', 'r')
correctPrint = correct_file.read()
try:
import io
from contextlib import redirect_stdout
with io.StringIO() as buf, redirect_stdout(buf):
print('redirected')
output = buf.getvalue()
self.assertEqual(output, correctPrint)
except ImportError:
import sys
f_handler = open('test.out', 'w')
sys.stdout=f_handler
self.table.create_table()
f_handler.close()
f_handler = open('test.out', 'r')
self.assertEqual(f_handler.read(), correctPrint)
class PylsyTableTests(unittest.TestCase):
def setUp(self):
attributes = ["name", "age"]
self.table = PylsyTable(attributes)
def tearDown(self):
self.table = None
def testCreateTable(self):
name = ["a", "b"]
self.table.add_data("name", name)
age = [1, 2]
self.table.add_data("age", age)
correct_file = open('./pylsy/tests/correct.out', 'r')
correctPrint = correct_file.read()
try:
import io
from contextlib import redirect_stdout
with io.StringIO() as buf, redirect_stdout(buf):
print('redirected')
output = buf.getvalue()
self.assertEqual(output, correctPrint)
except ImportError:
import sys
f_handler = open('test.out', 'w')
sys.stdout = f_handler
self.table.create_table()
f_handler.close()
f_handler = open('test.out', 'r')
self.assertEqual(f_handler.read(), correctPrint)
def setUp(self):
attributes=["name","age"]
self.table = PylsyTable(attributes)
def setUp(self):
attributes = ["name", "age"]
self.table = PylsyTable(attributes)