def test_get_size(self): # Create pty from standard stdin/stdout p = Pty() # Test get_size -> returns height,width size = p.get_size() self.assertIsInstance(size, tuple) self.assertEqual(len(size), 2) self.assertIsInstance(size[0], int) self.assertIsInstance(size[1], int) # Test get_width width = p.get_width() self.assertIsInstance(width, int) self.assertEqual(width, size[1]) # Test get_height height = p.get_height() self.assertIsInstance(height, int) self.assertEqual(height, size[0]) # Test set_term_var/get_term_var p.set_term_var('my-custom-xterm') self.assertEqual('my-custom-xterm', p.get_term_var())