Project Files

.pydev-tools

_ast

describe.py

do_work.py

docstrings.py

extract_function.py

extract.py

legacy_handler_2.py

legacy_handler_3.py

legacy_handler.py

migration.py

reference.py

rename.py

workspace_description.md

prompt_examples

dependency_version_migration.txt

hotfix_rework.txt

new_project.txt

performance_optimization.txt

refactory.txt

security_hardening.txt

source_tree_analysis.txt

test_coverage_expansion.txt

scripts

e2e_runPytest_test.cjs

setupVenv.mjs

smokeNew6Tools.ts

smokeNewTools.ts

verify_testfile_fix.ts

verifySetupVenv.ts

skills

pydev-mcp.md

src

tools

_profiling_worker_main.py

analyzeImportsAndDependencies.ts

buildPythonPackage.ts

checkForBugs.ts

checkForBugsInFile.ts

createProjectStructure.ts

createTestFileTool.ts

editTextFile.ts

editTextFileByLine.ts

findFiles.ts

generateRequirementsTxtTool.ts

getSetupStatusTool.ts

getWorkspaceRootTool.ts

inspectEnvironmentTool.ts

installModule.ts

listDirectory.ts

listTools.ts

profileProject.ts

pydevAuditDocstringsTool.ts

pydevDescribeWorkspaceTool.ts

pydevExtractFunctionTool.ts

pydevGenerateReferenceTool.ts

pydevMigrationAuditTool.ts

pydevSafeRenameTool.ts

readTextFile.ts

runLinterAndFormatter.ts

runLinterAndFormatterInFile.ts

runProjectCoverage.ts

runPython.ts

runPythonFile.ts

runPythonFileInteractive.ts

runPythonInteractive.ts

runRepl.ts

runTests.ts

runWithDebuggerTool.ts

saveTextFile.ts

scanProjectSecurity.ts

searchDirectory.ts

setupVenvTool.ts

switchPythonVersion.ts

typeCheckProjectTool.ts

uninstallModule.ts

config.ts

index.ts

test

run_tests.cjs

tools.test.cjs

manifest.json

package-lock.json

package.json

professor_headache.jpg

README.md

setup-venv.sh

tsconfig.json

src / tools / createTestFileTool.ts

import { tool } from "@lmstudio/sdk";
import { z } from "zod";

import { createTestFile, type TestStyle } from "../utils/createTestFileUtil";
import { getToolDefinition } from "../utils/toolRegistry";

const TOOL_DEFINITION = getToolDefinition("pydev_create_test_file")!;

export const createTestFileTool = tool({
  name: TOOL_DEFINITION.name,
  description: `${TOOL_DEFINITION.description} Generates a pytest test-file skeleton (fixture imports + assertion patterns) for an existing .py file.`,
  parameters: {
    targetFilePath: z
      .string()
      .min(1)
      .max(500)
      .describe("Path to the .py file you want tests generated for."),
    testFilePath: z
      .string()
      .min(1)
      .max(500)
      .optional()
      .describe("Where to write the generated test file. Defaults to tests/test_<stem>.py."),
    style: z
      .enum(["assert", "raises"])
      .optional()
      .describe("Assertion style for the generated skeleton. Defaults to 'assert'."),
    includeFixtures: z
      .boolean()
      .optional()
      .describe("Include an example pytest fixture in the skeleton. Defaults to true."),
  },
  implementation: async ({ targetFilePath, testFilePath, style, includeFixtures }) => {
    const result = await createTestFile(targetFilePath, {
      testFilePath,
      style: (style ?? "assert") as TestStyle,
      includeFixtures,
    });

    return {
      testFilePath: result.testFilePath,
      targetFilePath: result.targetFilePath,
      functions: result.functions,
      classes: result.classes,
      generatedTests: result.generatedTests,
      contentLength: result.contentLength,
    };
  },
});