Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix tests to use DataSource.from_proto
Signed-off-by: Felix Wang <wangfelix98@gmail.com>
  • Loading branch information
felixwang9817 committed Apr 23, 2022
commit 358799375bea56cf43a741365709ecfcc6db0ff5
9 changes: 5 additions & 4 deletions sdk/python/tests/unit/test_data_sources.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from feast import ValueType
from feast.data_format import ProtoFormat
from feast.data_source import (
DataSource,
KafkaSource,
KinesisSource,
PushSource,
Expand Down Expand Up @@ -195,8 +196,8 @@ def test_proto_conversion():
)

assert (
BigQuerySource.from_proto(bigquery_source.to_proto()) == bigquery_source_proto
DataSource.from_proto(bigquery_source.to_proto()) == bigquery_source
)
assert FileSource.from_proto(file_source.to_proto()) == file_source
assert RedshiftSource.from_proto(redshift_source.to_proto()) == redshift_source
assert SnowflakeSource.from_proto(snowflake_source.to_proto()) == snowflake_source
assert DataSource.from_proto(file_source.to_proto()) == file_source
assert DataSource.from_proto(redshift_source.to_proto()) == redshift_source
assert DataSource.from_proto(snowflake_source.to_proto()) == snowflake_source