Merge pull request #16 from intake/small_fix

Just because some sources use old style discover, should pass
intake · Nov 11, 2019 · e3eab2a · e3eab2a
2 parents e38d720 + 359b606
commit e3eab2a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/intake_parquet/__init__.py b/intake_parquet/__init__.py
@@ -3,4 +3,5 @@
 __version__ = get_versions()['version']
 del get_versions
 
+import intake
 from intake_parquet.source import ParquetSource
diff --git a/tests/test_discovery.py b/tests/test_discovery.py
@@ -8,4 +8,5 @@ def test_discovery():
     # For awhile we expect a PendingDeprecationWarning due to
     # do_pacakge_scan=True. But we should *not* get a FutureWarning.
     for record in record.list:
-        assert not isinstance(record.message, FutureWarning)
+        if isinstance(record.message, FutureWarning):
+            assert "parquet" not in str(record.message)