def test_sort_1(self): with patch('student_management_system.open', self.m_open1): sms = StudentManagementSystem() sms.read('score.csv') result = sms.sort(order_key="register", order_way="asc") self.assertEqual('1,강호민,85,90,95,270,90', result)
def test_read_2(self): with patch('student_management_system.open', self.m_open2): sms = StudentManagementSystem() self.assertEqual(2, sms.read('score.csv')) self.m_open2.assert_called_with('score.csv', 'rt', encoding="utf-8")
def test_sort_5(self): with patch('student_management_system.open', self.m_open2): sms = StudentManagementSystem() sms.read('score.csv') result = sms.sort("stotal", "des") self.assertEqual('1,강호민,85,90,95,270,90\n2,김광호,80,70,60,210,70', result)
def test_sort_3(self): with patch('student_management_system.open', self.m_open2): sms = StudentManagementSystem() sms.read('score.csv') result = sms.sort(order_key="register", order_way="des") self.assertEqual('2,김광호,80,70,60,210,70\n1,강호민,85,90,95,270,90', result)
def test_write_2(self): with patch('student_management_system.open', self.m_write_open_1): sms = StudentManagementSystem() sms.read('score.csv') sms.write('result.csv', 'register', 'des') self.m_w.write.assert_called_with( "3,김민식,75,85,80,240,80\n2,김광호,80,70,60,210,70\n1,강호민,85,90,95,270,90" )
from student_management_system import StudentManagementSystem if __name__ == "__main__": sms = StudentManagementSystem() sms.read('score.csv') sms.write('pre_result.csv', "stotal", "des") sms2 = StudentManagementSystem() sms2.read_2('pre_result.csv') print(sms2.sort_2("stotal", "des")) sms2.write_2('result.csv', "stotal", "des")
def test_constructor(self): sms = StudentManagementSystem() self.assertIsNotNone(sms)