Merge branch 'nedbat/data-sqlite'

1 files changed, 5 insertions, 6 deletions

diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py
index f4fbdbbb..b12f92ea 100644
--- a/tests/test_cmdline.py
+++ b/tests/test_cmdline.py
@@ -16,7 +16,7 @@ import coverage
 import coverage.cmdline
 from coverage import env
 from coverage.config import CoverageConfig
-from coverage.data import CoverageData, CoverageDataFiles
+from coverage.data import CoverageData
 from coverage.misc import ExceptionDuringRun
 
 from tests.coveragetest import CoverageTest, OK, ERR, command_line
@@ -585,8 +585,7 @@ class CmdLineWithFilesTest(BaseCmdLineTest):
             "file2.py": dict.fromkeys(range(1, 24)),
         })
         data.add_file_tracers({"file1.py": "a_plugin"})
-        data_files = CoverageDataFiles()
-        data_files.write(data)
+        data.write()
 
         self.command_line("debug data")
         self.assertMultiLineEqual(self.stdout(), textwrap.dedent("""\
@@ -597,16 +596,16 @@ class CmdLineWithFilesTest(BaseCmdLineTest):
             2 files:
             file1.py: 17 lines [a_plugin]
             file2.py: 23 lines
-            """).replace("FILENAME", data_files.filename))
+            """).replace("FILENAME", data.filename))
 
     def test_debug_data_with_no_data(self):
-        data_files = CoverageDataFiles()
+        data = CoverageData()
         self.command_line("debug data")
         self.assertMultiLineEqual(self.stdout(), textwrap.dedent("""\
             -- data ------------------------------------------------------
             path: FILENAME
             No data collected
-            """).replace("FILENAME", data_files.filename))
+            """).replace("FILENAME", data.filename))
 
 
 class CmdLineStdoutTest(BaseCmdLineTest):