def test_file(self):
with tempfile.TemporaryDirectory() as dir:
filename = os.path.join(dir, 'tests.py')
with open(filename, 'w') as fout:
fout.write(SAMPLE_DOCTEST)
result = doctests.testfile(filename, module_relative=False, report=False)
self.assertEqual(result.attempted, 3)
self.assertEqual(result.failed, 0)
def test_file_uses_beam_dataframes(self):
with tempfile.TemporaryDirectory() as dir:
filename = os.path.join(dir, 'tests.py')
with open(filename, 'w') as fout:
fout.write(CHECK_USES_DEFERRED_DATAFRAMES)
result = doctests.testfile(filename, module_relative=False, report=False)
self.assertNotEqual(result.attempted, 0)
self.assertEqual(result.failed, 0)
