Пример #1
0
    def test_simple(self):
        args = ['examples/bad.csv']
        output_file = six.StringIO()

        utility = CSVClean(args, output_file)
        utility.main()

        self.assertTrue(os.path.exists('examples/bad_err.csv'))
        self.assertTrue(os.path.exists('examples/bad_out.csv'))

        try:
            with open('examples/bad_err.csv') as f:
                next(f)
                self.assertEqual(next(f)[0], '1')
                self.assertEqual(next(f)[0], '2')
                self.assertRaises(StopIteration, next, f)
                
            with open('examples/bad_out.csv') as f:
                next(f)
                self.assertEqual(next(f)[0], '0')
                self.assertRaises(StopIteration, next, f)
        finally:
            # Cleanup
            os.remove('examples/bad_err.csv')
            os.remove('examples/bad_out.csv')
Пример #2
0
    def test_simple(self):
        args = ['examples/bad.csv']
        output_file = StringIO()

        utility = CSVClean(args, output_file)
        utility.main()

        self.assertTrue(os.path.exists('examples/bad_err.csv'))
        self.assertTrue(os.path.exists('examples/bad_out.csv'))

        try:
            with open('examples/bad_err.csv') as f:
                f.next()
                self.assertEqual(f.next()[0], '1')
                self.assertEqual(f.next()[0], '2')
                self.assertRaises(StopIteration, f.next)

            with open('examples/bad_out.csv') as f:
                f.next()
                self.assertEqual(f.next()[0], '0')
                self.assertRaises(StopIteration, f.next)
        finally:
            # Cleanup
            os.remove('examples/bad_err.csv')
            os.remove('examples/bad_out.csv')
Пример #3
0
    def test_dry_run(self):
        args = ['-n', 'examples/bad.csv']
        output_file = six.StringIO()

        utility = CSVClean(args, output_file)
        utility.main()

        self.assertFalse(os.path.exists('examples/bad_err.csv'))
        self.assertFalse(os.path.exists('examples/bad_out.csv'))

        output = six.StringIO(output_file.getvalue())

        self.assertEqual(next(output)[:6], 'Line 1')
        self.assertEqual(next(output)[:6], 'Line 2')
Пример #4
0
    def test_dry_run(self):
        args = ['-n', 'examples/bad.csv']
        output_file = six.StringIO()

        utility = CSVClean(args, output_file)
        utility.main()

        self.assertFalse(os.path.exists('examples/bad_err.csv'))
        self.assertFalse(os.path.exists('examples/bad_out.csv'))

        output = six.StringIO(output_file.getvalue())

        self.assertEqual(next(output)[:6], 'Line 1')
        self.assertEqual(next(output)[:6], 'Line 2')
Пример #5
0
def launch_new_instance():
    utility = CSVClean()
    utility.main()