fix(test): update tests for entity class refactoring#623

gkorland merged 1 commit intostagingFalkorDB/code-graph:stagingfrom

fix/broken-test-importsFalkorDB/code-graph:fix/broken-test-importsCopy head branch name to clipboard